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])