def test_arity_zero(self): self.assertEqual(xf.drop(1)(lambda: 'success')(), 'success')
def test_complete(self): xform = xf.comp(xf.drop(2), xf.partition_all(2)) dropped = list(xf.xiter(xform, [1, 2, 3, 4, 5])) self.assertEqual(list(dropped), [(3, 4), (5,)])
def test_reduced(self): xform = xf.comp(xf.drop(2), xf.take(2)) dropped = list(xf.xiter(xform, range(8))) self.assertEqual(dropped, [2, 3])
def test_drop_neg(self): dropped = list(xf.xiter(xf.drop(-1), [1, 2, 3, 4])) self.assertEqual(dropped, [1, 2, 3, 4])
def test_drop_zero(self): dropped = list(xf.xiter(xf.drop(0), [1, 2, 3, 4])) self.assertEqual(dropped, [1, 2, 3, 4])
def test_drop_too_many(self): dropped = list(xf.xiter(xf.drop(10), [1, 2, 3, 4])) self.assertEqual(dropped, [])
def test_drop_pos(self): dropped = list(xf.xiter(xf.drop(2), [1, 2, 3, 4])) self.assertEqual(dropped, [3, 4])