Класс TestGetScore тестирующий функцию get_score()

test_getscore.py

from unittest import TestCase, main
from pyshop import get_score


class TestGetScore(TestCase):

    game_stamps = [{'offset': 100043, 'score': {'away': 2, 'home': 4}},
                   {'offset': 100044, 'score': {'away': 2, 'home': 4}},
                   {'offset': 100045, 'score': {'away': 100, 'home': 100}},
                   {'offset': 100047, 'score': {'away': 2, 'home': 4}},
                   {'offset': 100048, 'score': {'away': 10, 'home': 14}},
                   {'offset': 100050, 'score': {'away': 2, 'home': 4}}]

    def test_offset_value_not_in_offset_range(self):
        self.assertRegex('Вне диапазона', get_score(self.game_stamps, 999))

    def test_offset_value_must_be_int(self):
        self.assertRegex('"offset" должен быть целым числом', get_score(self.game_stamps, 'eee'))

    def test_get_score_value(self):
        self.assertEqual(get_score(self.game_stamps, 100048), 'away: 10, home: 14')

    # "closest to the left" case
    def test_offset_in_offset_range_but_not_in_offset_list(self):
        self.assertEqual(get_score(self.game_stamps, 100049), 'away: 10, home: 14')


if __name__ == '__main__':
    main()