Subversion Repositories Local Hare Voting

Rev

Go to most recent revision | Details | 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):
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 
 
63 jtkorb 21 
	tally.fParallel = 1
22 
 
57 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)
26 
 
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 
 
61 jtkorb 31 
	self.assertEqual(x.forkPendingRedist(["A"], "A"), "A")
32 
	self.assertRaises(ZeroDivisionError, x.forkPendingRedist, [], None)
57 jtkorb 33 
 
34 
    def testCombinations(self):
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)
40 
 
41 
    def doCombination(self, l, n, t):
42 
	c = 0
43 
	for i in tally.combinations(l, n):
44 
	    self.assertEqual(len(i), n)
61 jtkorb 45 
	    print i
57 jtkorb 46 
	    c += 1
47 
	self.assertEqual(c, t)
48 
 
15 jtkorb 49 
if __name__=="__main__":
50 
    unittest.main()