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])]