def test_collect():
a = PBag(identity, 2)
with a:
a.extend([0, 1, 2, 3])
b = PBag(identity, 2)
with b:
b.extend([0, 1, 2, 3])
result = merge(dict(collect(identity, 2, 0, [a, b])),
dict(collect(identity, 2, 1, [a, b])))
assert result == {0: [0, 0], 1: [1, 1], 2: [2, 2], 3: [3, 3]}
def test_collect():
a = PBag(identity, 2)
with a:
a.extend([0, 1, 2, 3])
b = PBag(identity, 2)
with b:
b.extend([0, 1, 2, 3])
result = merge(dict(collect(identity, 2, 0, [a, b])),
dict(collect(identity, 2, 1, [a, b])))
assert result == {0: [0, 0],
1: [1, 1],
2: [2, 2],
3: [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])]
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])]
