def test_complete(self): xform = xf.comp(xf.drop_while(lambda x: x < 3), xf.partition_all(2)) dropped = list(xf.xiter(xform, range(8))) self.assertEqual(list(dropped), [(3, 4), (5, 6), (7,)])
def test_arity_zero(self): self.assertEqual(xf.drop_while(None)(lambda: 'success')(), 'success')
def test_pred_ignored_after_first_take(self): dropped = list(xf.xiter(xf.drop_while(lambda x: x < 3), [1, 2, 3, -4, -5])) self.assertEqual(dropped, [3, -4, -5])
def test_reduced(self): xform = xf.comp(xf.drop_while(lambda x: x < 3), xf.take(2)) dropped = list(xf.xiter(xform, range(8))) self.assertEqual(list(dropped), [3, 4])
def test_drop_none(self): dropped = list(xf.xiter(xf.drop_while(lambda x: x < 0), [1, 2, 3, 4])) self.assertEqual(dropped, [1, 2, 3, 4])