def test_pop_ids_too_many(self):
five = set(range(5))
back = pop_ids(five, 6)
self.assertEqual(5, len(back))
self.assertEqual(0, len(five))
self.assertEqual(set(back), set(range(5)))
def test_pop_ids_too_many(self):
five = set(range(5))
back = pop_ids(five, 6)
self.assertEqual(5, len(back))
self.assertEqual(0, len(five))
self.assertEqual(set(back), set(range(5)))
def test_pop_ids(self):
five = set(range(5))
three = pop_ids(five, 3)
self.assertEqual(3, len(three))
self.assertEqual(2, len(five))
self.assertEqual(five | set(three), set(range(5)))
def test_pop_ids(self):
five = set(range(5))
three = pop_ids(five, 3)
self.assertEqual(3, len(three))
self.assertEqual(2, len(five))
self.assertEqual(five | set(three), set(range(5)))
