def test_Ballot_class6(self): ballot = Ballot([5,6,7]) self.assertEqual(ballot.get_next(2),7)
def test_Ballot_class3(self): ballot = Ballot([1,2,3]) self.assertEqual(ballot.get_next(2),3)