def test_emptyRange(self): self.assertEqual(list(iterage.drop(range(6), 0)), list(range(6))) self.assertEqual(list(iterage.drop([], 3)), []) self.assertEqual(list(iterage.drop([], 0)), [])
def test_simpleUseCases(self): for i in range(6): exp = [] if i < 3 else list(range(3, i)) self.assertEqual(list(iterage.drop(range(i), 3)), exp) self.assertEqual(list(iterage.drop(list(range(i)), 3)), exp) self.assertEqual(list(iterage.drop(createNGenerator(i), 3)), exp)