Exemplo n.º 1
0
 def test_remove(self):
     """
     Removing things from the list should remove them immediately from the
     cycle.
     """
     ch = RoundRobinChooser()
     ch.add('a')
     ch.add('b')
     ch.add('c')
     self.assertEqual(ch.next(), 'a')
     self.assertEqual(ch.next(), 'b')
     ch.remove('c')
     self.assertEqual(ch.next(), 'a')
     ch.remove('a')
     self.assertEqual(ch.next(), 'b')
     self.assertEqual(ch.next(), 'b')
Exemplo n.º 2
0
 def test_basic(self):
     """
     You can add things to the list and cycle through them.
     """
     ch = RoundRobinChooser()
     ch.add('a')
     ch.add('b')
     ch.add('c')
     self.assertEqual(ch.next(), 'a')
     self.assertEqual(ch.next(), 'b')
     self.assertEqual(ch.next(), 'c')
     self.assertEqual(ch.next(), 'a')
     ch.add('d')
     self.assertEqual(ch.next(), 'b')
     self.assertEqual(ch.next(), 'c')
     self.assertEqual(ch.next(), 'd')