fixed small errors in unit testing

This commit is contained in:
Brandon Goodell 2017-12-13 10:53:39 -07:00
parent 1445943246
commit fef1017fb7

View file

@ -417,7 +417,7 @@ class Test_RoBlock(unittest.TestCase):
R.voteFor((gcid,xid,yid),{}) R.voteFor((gcid,xid,yid),{})
self.assertEqual(R.votes[(gcid,xid,yid)],1) self.assertEqual(R.votes[(gcid,xid,yid)],1)
self.assertEqual(R.votes[(gcid,yid,xid)],-1) self.assertEqual(R.votes[(gcid,yid,xid)],-1)
for pid in R.blocks[vid].parents: for pid in R.blocks[gcid].parents:
if pid in R.vids: if pid in R.vids:
q.append(gpid) q.append(gpid)
while(len(q)>0): while(len(q)>0):
@ -432,13 +432,13 @@ class Test_RoBlock(unittest.TestCase):
R.voteFor((gcid,yid,xid),{}) R.voteFor((gcid,yid,xid),{})
self.assertEqual(R.votes[(gcid,xid,yid)],-1) self.assertEqual(R.votes[(gcid,xid,yid)],-1)
self.assertEqual(R.votes[(gcid,yid,xid)],1) self.assertEqual(R.votes[(gcid,yid,xid)],1)
for pid in R.blocks[vid].parents: for pid in R.blocks[gcid].parents:
if pid in R.vids: if pid in R.vids:
q.append(gpid) q.append(gpid)
while(len(q)>0): while(len(q)>0):
wid = q.popleft() wid = q.popleft()
self.assertEqual(R.pendingVotes[(wid,xid,yid)],0) self.assertEqual(R.pendingVotes[(wid,xid,yid)],0)
self.assertEqual(R.pendingVotes[(wid,yid,yid)],0) self.assertEqual(R.pendingVotes[(wid,yid,xid)],0)
for pid in R.blocks[wid].parents: for pid in R.blocks[wid].parents:
if pid in R.vids: if pid in R.vids:
q.append(pid) q.append(pid)
@ -446,18 +446,23 @@ class Test_RoBlock(unittest.TestCase):
R.voteFor((gcid,yid,xid),{}) R.voteFor((gcid,yid,xid),{})
self.assertEqual(R.votes[(gcid,xid,yid)],-1) self.assertEqual(R.votes[(gcid,xid,yid)],-1)
self.assertEqual(R.votes[(gcid,yid,xid)],1) self.assertEqual(R.votes[(gcid,yid,xid)],1)
for pid in R.blocks[vid].parents: for pid in R.blocks[gcid].parents:
if pid in R.vids: if pid in R.vids:
q.append(gpid) q.append(gpid)
while(len(q)>0): while(len(q)>0):
wid = q.popleft() wid = q.popleft()
self.assertEqual(R.pendingVotes[(wid,xid,yid)],-1) self.assertEqual(R.pendingVotes[(wid,xid,yid)],-1)
self.assertEqual(R.pendingVotes[(wid,yid,yid)],-1) self.assertEqual(R.pendingVotes[(wid,yid,xid)],1)
for pid in R.blocks[wid].parents: for pid in R.blocks[wid].parents:
if pid in R.vids: if pid in R.vids:
q.append(pid) q.append(pid)
# Test sumPendingVotes
R.sumPendingVote(gcid, {})
self.assertTrue((gcid,xid,yid) in R.votes)
self.assertTrue((gcid,yid,xid) in R.votes)
self.assertEqual(R.votes[(gcid,xid,yid)],-1)
self.assertEqual(R.votes[(gcid,yid,xid)],1)