def test_compose():
def f(it):
for x in it:
yield x+1
g = dlf.compose(f, f, f)
assert list(g(range(5))) == [3, 4, 5, 6, 7]
reload(dlf)
def f(s):
def g(it):
for x in it:
yield x+s
return g
g = dlf.compose(f("a"), f("b"), f("c"), f("d"))
assert list(g([""])) == ["abcd"]
def test_compose():
for sample in dlf.compose(dlf.batched(20), dlf.unbatch())(source2(100)):
assert sample["png"].shape == (1, 1, 3), sample["png"].shape