def test_head(): assert list(compute(t.head(1), data)) == [data[0]] e = head(t, 101) p = list(range(1000)) assert len(list(compute(e, p))) == 101