source: geniuswebcore/test/geniusweb/voting/votingevaluators/LargestAgreementTest.py@ 72

Last change on this file since 72 was 72, checked in by Bart Vastenhouw, 3 years ago

Reduced memory need of websockets.

File size: 2.4 KB
Line 
1import json
2from typing import List
3import unittest
4from unittest.mock import Mock
5
6from pyson.ObjectMapper import ObjectMapper
7from unitpy.GeneralTests import GeneralTests
8
9from geniusweb.actions.PartyId import PartyId
10from geniusweb.actions.Vote import Vote
11from geniusweb.actions.Votes import Votes
12from geniusweb.issuevalue.Bid import Bid
13from geniusweb.voting.CollectedVotes import CollectedVotes
14from geniusweb.voting.VotingEvaluator import VotingEvaluator
15from geniusweb.voting.votingevaluators.LargestAgreement import LargestAgreement
16
17
18class LargestAgreementTest (unittest.TestCase, GeneralTests[LargestAgreement]):
19 jackson = ObjectMapper()
20
21 lal1 = LargestAgreement()
22 lal1a = LargestAgreement()
23 party1 = PartyId("party1")
24 party2 = PartyId("party2")
25 party3 = PartyId("party3")
26 party4 = PartyId("party4")
27 votes1 = Mock(Votes)
28 lal2 = LargestAgreement()
29 # hack for testing only.
30 lal2._allVotes = (CollectedVotes({party1: votes1}, {party1: 1}))
31
32 lalstring = "{\"LargestAgreement\":{}}"
33
34 def getGeneralTestData(self)-> List[List[LargestAgreement]] :
35 return [[self.lal1, self.lal1a], [self.lal2]]
36
37 def getGeneralTestStrings(self) ->List[str] :
38 return ["LargestAgreement", "LargestAgreement"]
39
40 def testserializeTest(self) :
41 # we use lal2 to also check we don't serialize the contents
42 print(self.jackson.toJson(self.lal2))
43 self.assertEqual(json.loads(self.lalstring), self.jackson.toJson(self.lal2))
44
45 def testdeserializeTest(self):
46 eval = self.jackson.parse(json.loads(self.lalstring),
47 VotingEvaluator)
48 # it should deserialize as the empty lal1 because we ignore the fields.
49 self.assertEqual(self.lal1, eval)
50
51 def testCollectVotes(self):
52 # we set up a tricky situation where party4 has much more power
53 # and therefore party 1+4 >> p1+2+3
54 bid1 = Mock(Bid)
55 bid2 = Mock(Bid)
56
57 votes = CollectedVotes({}, {})
58 votes = votes.With(Votes(self.party1,
59 set([Vote(self.party1, bid1, 1, 99),
60 Vote(self.party1, bid2, 1, 99)])),
61 1)
62 votes = votes.With(
63 Votes(self.party2,
64 set([Vote(self.party2, bid1, 1, 99)])),
65 1)
66 votes = votes.With(
67 Votes(self.party3,
68 set([Vote(self.party3, bid1, 1, 99)])),
69 1)
70 votes = votes.With(
71 Votes(self.party4,
72 set([Vote(self.party4, bid2, 1, 99)])),
73 4)
74 la = LargestAgreement()
75 la._allVotes=votes
76 agreement = la._collectVotes()
77 self.assertEqual(set([self.party1, self.party4]),
78 agreement.getMap().keys())
Note: See TracBrowser for help on using the repository browser.