def test_partition_set6():
stuff = ['0', '1', '00', '11', '000', '111', (0, 1, 2)]
fn = lambda a, b: a[0] == b[0]
eqclasses, _ = partition_set(stuff,
fn,
reflexive=True,
transitive=False,
innerset=True)
assert eqclasses == [
frozenset(['0', '00', '000']),
frozenset(['1', '11', '111']),
frozenset([(0, 1, 2)])
]
def test_partition_set4():
stuff = ['01', '10', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: a[0] == b[1]
_, lookup = partition_set(stuff, fn, reflexive=False, transitive=True)
assert lookup == [0, 0, 1, 2, 1, 2, 3]
def test_partition_set3():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
_, lookup = partition_set(stuff, reflexive=False, transitive=True)
assert lookup == [0, 1, 2, 3, 4, 5, 6]
def test_partition_set2():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: len(a) == len(b)
_, lookup = partition_set(stuff, fn, reflexive=True, transitive=True)
assert lookup == [0, 0, 1, 1, 2, 2, 2]
def test_partition_set1():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: len(a) == len(b)
_, lookup = partition_set(stuff, fn)
assert lookup == [0, 0, 1, 1, 2, 2, 2]
def test_partition_set2():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: len(a) == len(b)
_, lookup = partition_set(stuff, fn, reflexive=True, transitive=True)
assert_equal(lookup, [0, 0, 1, 1, 2, 2, 2])
def test_partition_set1():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: len(a) == len(b)
_, lookup = partition_set(stuff, fn)
assert_equal(lookup, [0, 0, 1, 1, 2, 2, 2])
def test_partition_set4():
stuff = ['01', '10', '00', '11', '000', '111', [0, 1, 2]]
fn = lambda a, b: a[0] == b[1]
_, lookup = partition_set(stuff, fn, reflexive=False, transitive=True)
assert lookup == [0, 0, 1, 2, 1, 2, 3]
def test_partition_set3():
stuff = ['0', '1', '00', '11', '000', '111', [0, 1, 2]]
_, lookup = partition_set(stuff, reflexive=False, transitive=True)
assert lookup == [0, 1, 2, 3, 4, 5, 6]
def test_partition_set6():
stuff = ['0', '1', '00', '11', '000', '111', (0, 1, 2)]
fn = lambda a, b: a[0] == b[0]
eqclasses, _ = partition_set(stuff, fn, reflexive=True, transitive=False, innerset=True)
assert eqclasses == [frozenset(['0', '00', '000']), frozenset(['1', '11', '111']), frozenset([(0, 1, 2)])]
