def test_partition_collect(): with partd.Pickle() as p: partition(identity, range(6), 3, p) assert set(p.get(0)) == {0, 3} assert set(p.get(1)) == {1, 4} assert set(p.get(2)) == {2, 5} assert sorted(collect(identity, 0, p, "")) == [(0, [0]), (3, [3])]
def test_partition_collect(): with partd.Pickle() as p: partition(identity, range(6), 3, p) assert set(p.get(0)) == set([0, 3]) assert set(p.get(1)) == set([1, 4]) assert set(p.get(2)) == set([2, 5]) assert sorted(collect(identity, 0, p, '')) == [(0, [0]), (3, [3])]