bestRouteElectionTool/test_election.py at master · jochemhoorneman/bestRouteElectionTool · 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
from unittest import TestCase
from random import choices, sample
import string

from election import ElectionResult, find_irv_winner, get_smith_set, find_tideman_winner
from voter import Voter


def get_random_voter_on(routes):
    random_name = ''.join(choices(string.ascii_uppercase + string.digits, k=7))
    return Voter(random_name, sample(routes, k=5))


class TestElectionResult(TestCase):
    def setUp(self) -> None:
        num_routes = 7
        routes = ['r{}'.format(i) for i in range(1, num_routes + 1)]
        self.test_election = ElectionResult(
            routes, [
                Voter('Ike', ['r1', 'r6', 'r2', 'r3', 'r5', 'r4', 'r7']),
                Voter('Jochem', ['r3', 'r4', 'r6', 'r1', 'r2', 'r7', 'r5']),
                Voter('Jesse', ['r2', 'r1', 'r5', 'r3', 'r4', 'r7', 'r6']),
                Voter('Piotr', ['r3', 'r5', 'r2', 'r4', 'r7', 'r1', 'r6']),
                Voter('Melina', ['r4', 'r1', 'r5', 'r6', 'r3', 'r2', 'r7']),
            ]
        )

    def test_get_winner(self):
        self.assertIsNone(self.test_election.get_winner())
        self.assertIsNone(self.test_election.without_route('r2').get_winner())
        self.assertEqual(self.test_election.without_route('r2').without_route('r4').get_winner(), 'r1')

    def test_rank_routes(self):
        self.assertDictEqual(self.test_election.rank_routes(), {
            'r1': 1, 'r2': 1, 'r3': 2, 'r4': 1, 'r5': 0, 'r6': 0, 'r7': 0
        })

    def test_find_last_place(self):
        # TODO: add support for tie in last place
        print(self.test_election.get_interim_score())
        self.assertIn(self.test_election.find_last_place(), ['r5', 'r6', 'r7'])

    def test_without_route(self):
        election_without = self.test_election.without_route('r2')
        self.assertDictEqual(election_without.rank_routes(), {
            'r1': 2, 'r3': 2, 'r4': 1, 'r5': 0, 'r6': 0, 'r7': 0
        })
        # check that test_election is unchanged
        self.test_rank_routes()
        # check that voter rankings are changed correctly : original votes, with r2 removed
        self.assertListEqual(election_without.voters[3].ranking, ['r3', 'r5', 'r4', 'r7', 'r1', 'r6'])

    def test_find_winner(self):
        self.assertEqual(find_irv_winner(self.test_election), 'r1')

    def test_tideman_winner(self):
        self.assertEqual(find_tideman_winner(self.test_election), 'r1')

    def test_pairwise_votes(self):
        votes = self.test_election.get_pairwise_votes()
        self.assertGreater(votes['r1']['r3'], votes['r3']['r1'])
        self.assertGreater(votes['r4']['r1'], votes['r1']['r4'])
        self.assertGreater(votes['r3']['r4'], votes['r4']['r3'])

    def test_smith_set(self):
        # We have r4 > r1, r3 > r4. and additionally, r2 > r4 > r1 > r2
        # Also r6 > r2 and r5 > r4, thus
        self.assertSetEqual(get_smith_set(self.test_election), {'r1', 'r2', 'r3', 'r4', 'r5', 'r6'})
        removed_some = self.test_election.without_route('r7').without_route('r6').without_route('r5')
        self.assertEqual(get_smith_set(removed_some), {'r1', 'r2', 'r3', 'r4'})
        # remaining choice order does not matter: r1 will win
        self.assertEqual(get_smith_set(removed_some.without_route('r2').without_route('r4')), {'r1'})
        self.assertEqual(get_smith_set(removed_some.without_route('r4').without_route('r2')), {'r1'})