def test_next(self): pq = PlayerQueue({1:'a', 2:'b', 3:'c', 4:'d'}) self.assertEqual(1, pq.next()) self.assertEqual(2, pq.next()) self.assertEqual(3, pq.next()) self.assertEqual(4, pq.next()) self.assertEqual(1, pq.next())
def test_setFirst(self): pq = PlayerQueue({1:'a', 2:'b', 3:'c', 4:'d'}) pq.setFirst(3) self.assertEqual(3, pq.next())