bemaniutils/bemani/tests/test_IIDXPendual.py

# vim: set fileencoding=utf-8
import struct
import unittest
import random
from unittest.mock import Mock
from typing import List

from bemani.backend.iidx.pendual import IIDXPendual
from bemani.data import Score


class TestIIDXPendual(unittest.TestCase):

    def __make_score(self, ghost: List[int]) -> Score:
        return Score(
            0,
            0,
            0,
            sum(ghost),
            1234567890,
            1234567890,
            0,
            1,
            {
                'ghost': bytes(ghost),
            },
        )

    def test_average_no_scores(self) -> None:
        base = IIDXPendual(Mock(), Mock(), Mock())
        self.assertEqual(
            base.delta_score([], 3),
            (None, None),
        )

    def test_average_identity(self) -> None:
        base = IIDXPendual(Mock(), Mock(), Mock())
        self.assertEqual(
            base.delta_score([
                self.__make_score([10, 20, 30]),
            ], 3),
            (60, struct.pack('bbb', *[-10, 0, 10])),
        )

    def test_average_basic(self) -> None:
        base = IIDXPendual(Mock(), Mock(), Mock())
        self.assertEqual(
            base.delta_score([
                self.__make_score([10, 20, 30]),
                self.__make_score([0, 0, 0]),
            ], 3),
            (30, struct.pack('bbb', *[-5, 0, 5])),
        )

    def test_average_complex(self) -> None:
        base = IIDXPendual(Mock(), Mock(), Mock())
        self.assertEqual(
            base.delta_score([
                self.__make_score([10, 20, 30]),
                self.__make_score([20, 30, 40]),
                self.__make_score([30, 40, 50]),
            ], 3),
            (90, struct.pack('bbb', *[-10, 0, 10])),
        )

    def test_average_always_zero(self) -> None:
        base = IIDXPendual(Mock(), Mock(), Mock())
        ex_score, ghost = base.delta_score([
            self.__make_score([random.randint(0, 10) for _ in range(64)]),
            self.__make_score([random.randint(0, 10) for _ in range(64)]),
        ], 64)
        self.assertEqual(sum(struct.unpack('b' * 64, ghost)), 0)