python-study/module12_2.py at main · dkochmashev/python-study · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import unittest
from runner_and_tournament import *


class FairTournament(Tournament):
    def start(self):
        finishers = {}
        place = 1
        by_speed = sorted(self.participants, key=lambda i: i.speed, reverse=True)
        while by_speed:
            for participant in by_speed:
                participant.run()
                if participant.distance >= self.full_distance:
                    finishers[place] = participant
                    place += 1
                    by_speed.remove(participant)
                    break

        return finishers


class TournamentTest(unittest.TestCase):
    all_results = list()

    def runTest(self):
        self.test_1()
        self.test_2()
        self.test_3()
        self.test_4()

    @classmethod
    def setUpClass(cls):
        TournamentTest.all_results = list()

    def setUp(self):
        self.runners = {
            'Усэйн': Runner('Усэйн', 10),
            'Андрей': Runner('Андрей', 9),
            'Ник': Runner('Ник', 3)
        }

    @classmethod
    def tearDownClass(cls):
        for result in TournamentTest.all_results:
            print(result)

    def __add_results__(self, results):
        TournamentTest.all_results.append({place: results[place].name for place in results})

    # Усэйн и Ник
    def test_1(self):
        tournament = Tournament(90, self.runners['Усэйн'], self.runners['Ник'])
        self.assertIsNotNone(results := tournament.start())
        self.assertEqual(len(results), 2)
        self.assertNotEqual(results[1], results[2])
        self.assertEqual(results[1].name, 'Усэйн')
        self.assertEqual(results[2].name, 'Ник')
        self.__add_results__(results)

    # Андрей и Ник
    def test_2(self):
        tournament = Tournament(90, self.runners['Андрей'], self.runners['Ник'])
        self.assertIsNotNone(results := tournament.start())
        self.assertEqual(len(results), 2)
        self.assertNotEqual(results[1], results[2])
        self.assertEqual(results[1].name, 'Андрей')
        self.assertEqual(results[2].name, 'Ник')
        self.__add_results__(results)

    # Усэйн, Андрей и Ник.
    def test_3(self):
        tournament = Tournament(90, self.runners['Усэйн'], self.runners['Андрей'], self.runners['Ник'])
        self.assertIsNotNone(results := tournament.start())
        self.assertEqual(len(results), 3)
        self.assertNotEqual(results[1], results[2])
        self.assertNotEqual(results[1], results[3])
        self.assertNotEqual(results[2], results[3])
        self.assertEqual(results[1].name, 'Усэйн')
        self.assertEqual(results[2].name, 'Андрей')
        self.assertEqual(results[3].name, 'Ник')
        self.__add_results__(results)

    # Ошибка в алгоритме:
    # ---
    # Если значение скорости бегуна (Runner.speed) совпадает с дистанцией (Tournament.full_distance)
    # и бегун стоит первым в списке participants при конструировании Tournament, он будет на первом
    # месте в результатах забега, не смотря на то, что его скорость может быть ниже, чем у остальных
    # бегунов.
    #
    # Для выявления ошибки создан производный класс FairTournament с переопределением метода start,
    # и сравниваются результаты работы методов start() в вышеописанных условиях.

    def test_4(self):
        tournament = Tournament(3, self.runners['Ник'], self.runners['Усэйн'], self.runners['Андрей'])
        fair_tournament = FairTournament(3, self.runners['Ник'], self.runners['Усэйн'], self.runners['Андрей'])
        self.assertIsNotNone(results := tournament.start())
        fair_results = fair_tournament.start()
        self.assertDictEqual(results, fair_results, 'Неверный подсчет результатов в Tournament.start()')