Subversion Repositories Local Hare Voting

Rev

Rev 75 | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
72 jtkorb 1 
#!/usr/bin/env python
15 jtkorb 2 
 
3 
import os
4 
import unittest
5 
import random
6 
 
54 jtkorb 7 
import tally
15 jtkorb 8 
 
9 
class ModTest(unittest.TestCase):
10 
    def testNormalCase(self):
21 jtkorb 11 
        self.assertEqual(1, 1)
15 jtkorb 12 
 
54 jtkorb 13 
    def testUniverse(self):
78 jtkorb 14 
        x = tally.Universe(1.0, 3, 100, quota=5)
15 
        self.assertEqual(x.p, 1.0)
16 
        self.assertEqual(x.id, 1)
17 
        self.assertEqual(x.nBallots, 100)
18 
        self.assertEqual(x.quota, 5)
19 
        self.assertEqual(x.winners, [])
57 jtkorb 20 
 
78 jtkorb 21 
        tally.fParallel = 1
63 jtkorb 22 
 
78 jtkorb 23 
        self.assertEqual(x.forkPendingWinners([5,1,2,3]), 5)
24 
        self.assertEqual(x.p, 1.0/4)
25 
        self.assertEqual(len(tally.Universe.pendingUniverses), 3)
57 jtkorb 26 
 
78 jtkorb 27 
        self.assertEqual(x.forkPendingLosers(["A",1,2,3,4,5]), "A")
28 
        self.assertEqual(x.p, 1.0/4.0/6)
29 
        self.assertEqual(len(tally.Universe.pendingUniverses), 8)
54 jtkorb 30 
 
78 jtkorb 31 
        self.assertEqual(x.forkPendingRedist(["A"], "A"), "A")
32 
        self.assertRaises(ZeroDivisionError, x.forkPendingRedist, [], None)
57 jtkorb 33 
 
34 
    def testCombinations(self):
78 jtkorb 35 
        l = ["A", "B", "C", "D", "E"]
36 
        self.doCombination(l, 2, 10)
37 
        self.doCombination(l, 3, 10)
38 
        self.doCombination(l, 4, 5)
39 
        self.doCombination(l, 5, 1)
57 jtkorb 40 
 
41 
    def doCombination(self, l, n, t):
78 jtkorb 42 
        c = 0
43 
        for i in tally.combinations(l, n):
44 
            self.assertEqual(len(i), n)
45 
            print i
46 
            c += 1
47 
        self.assertEqual(c, t)
57 jtkorb 48 
 
15 jtkorb 49 
if __name__=="__main__":
50 
    unittest.main()