def test_sets_size():
r = hd.Range(3)
assert hd.Subsets(r, 2).size == 3
assert hd.Subsets(r, 0).size == 1
s = hd.Subsets(r * r, 1, 3)
assert s.size == len(list(s))
assert s.size == 129
def test_sets_in_sets():
s = hd.Subsets(hd.Subsets(hd.Range(1), set_class=frozenset),
set_class=frozenset)
assert set(s) == {
frozenset(),
frozenset([frozenset()]),
frozenset([frozenset(), frozenset([0])]),
frozenset([frozenset([0])])
}
def test_sets_generate():
r = hd.Range(3)
s = hd.Subsets(r, 2, set_class=frozenset)
result = set(s.generate(100))
assert result == set(frozenset(s) for s in ((0, 1), (0, 2), (1, 2)))
r = hd.Range(3)
s = hd.Subsets(r, set_class=frozenset)
result = set(s.generate(160))
print(result)
assert result == set(
frozenset(s)
for s in ((), (0, ), (1, ), (2, ), (0, 1), (0, 2), (1, 2), (0, 1, 2)))
def test_sets_to_values():
r = hd.Range(4)
s = hd.Subsets(r)
v = s.to_values()
assert isinstance(v, hd.Values)
assert list(v) == list(s)
def test_sets_to_values_maxsize():
r = hd.Range(4)
s = hd.Subsets(r)
v = s.to_values(r.size)
assert isinstance(v, hd.Subsets)
assert isinstance(v.domain, hd.Values)
assert list(v) == list(s)
def test_sets_set_class():
r = hd.Range(2)
s = hd.Subsets(r, set_class=frozenset)
assert not s.strict
result = set(s)
expected = {
frozenset(),
frozenset((0, )),
frozenset((1, )),
frozenset((0, 1))
}
assert result == expected
def test_sets_iter():
r = hd.Range(2)
s = hd.Subsets(r, 0)
assert list(s) == [hd.Set(())]
s = hd.Subsets(r, 1)
assert list(s) == [hd.Set((0, )), hd.Set((1, ))]
s = hd.Subsets(r, 2)
assert list(s) == [hd.Set((0, 1))]
s = hd.Subsets(r, 3)
assert list(s) == []
r3 = hd.Range(3)
s = hd.Subsets(r3, 0, 1)
assert list(s) == [hd.Set(()), hd.Set((0, )), hd.Set((1, )), hd.Set((2, ))]
s = hd.Subsets(r3, 0, 2)
assert list(s) == [
hd.Set(()),
hd.Set((0, )),
hd.Set((0, 1)),
hd.Set((0, 2)),
hd.Set((1, )),
hd.Set((1, 2)),
hd.Set((2, ))
]
s = hd.Subsets(r3, 1, 2)
assert list(s) == [
hd.Set((0, )),
hd.Set((0, 1)),
hd.Set((0, 2)),
hd.Set((1, )),
hd.Set((1, 2)),
hd.Set((2, ))
]
def test_canonical_sets():
ax = USet(4, "a")
s = hd.Subsets(ax, 2)
bf_check(s)
s = hd.Subsets(ax, 0, 2)
bf_check(s)
def test_sets_repr():
r = hd.Range(10)
assert repr(hd.Subsets(r)) == \
"<Subsets size=1024 " \
"{{}, {0}, {0, 1}, {0, 1, 2}, {0, 1, 2, 3}, ...}>"
def test_sets_strict():
r = hd.Range(2)
s = hd.Subsets(r, 0)
assert s.strict
def test_sets_invalid():
r = hd.Range(3)
s = hd.Subsets(r, 10)
assert list(s) == []
def haydi_parallel_iterate(count):
nodes = hd.ASet(count, "n")
graphs = hd.Subsets(hd.Subsets(nodes, 2))
ctx = hd.DistributedContext(spawn_workers=8)
return list(graphs.iterate().run(ctx=ctx))
def haydi_cnfs(count):
nodes = hd.ASet(count, "n")
graphs = hd.Subsets(hd.Subsets(nodes, 2))
return list(graphs.cnfs())
