def test_complete(self):
xform = xf.comp(xf.keep(lambda x: x if x % 2 == 0 else None),
xf.partition_all(2))
self.assertEqual(list(xf.xiter(xform, [2, 4, 5, 6])), [(2, 4), (6,)])
def test_arity_zero(self):
self.assertEqual(xf.keep(None)(lambda: 'success')(), 'success')
def test_keep_empty(self):
xform = xf.keep(lambda x: x if x % 2 == 0 else None)
self.assertEqual(list(xf.xiter(xform, [])), [])
def test_reduced(self):
xform = xf.comp(xf.keep(lambda x: x if x % 2 == 0 else None),
xf.take(2))
self.assertEqual(list(xf.xiter(xform, [1, 2, 3, 4, 5, 6])), [2, 4])
def test_keep_some(self):
xform = xf.keep(lambda x: x if x % 2 == 0 else None)
self.assertEqual(list(xf.xiter(xform, [1, 2, 3, 4])), [2, 4])