def test_rims():
assert set(Partition.rims((), 0)) == {()}
assert set(Partition.rims((), 1)) == {(), ((1, 1), )}
assert set(Partition.rims((), 2)) == {(), ((1, 1), ), ((1, 1), (1, 2))}
assert set(Partition.rims((1, ), 1)) == {(), ((1, 2), ), ((1, 2), (2, 2))}
assert set(Partition.rims(
(3, 2))) == {(), ((1, 4), ), ((1, 4), (2, 4)), ((3, 3), ),
((1, 4), (3, 3)), ((1, 4), (2, 4), (3, 3)),
((1, 4), (2, 4), (3, 3), (3, 4))} # noqa
def KLG_counts_by_shape(cls, content_max, mu): # noqa
ans = {}
for rim in Partition.rims(mu, content_max + 1):
count = cls._KLG_count_helper(content_max, rim)
if count:
nu = cls._add_rim(mu, rim)
ans[nu] = count
return ans
def KLG(cls, content_max, mu): # noqa
for shape in Partition.rims(mu, content_max + 1):
for tab in cls._KLG_helper(content_max, (), shape):
yield Tableau({(i, j): a for (i, j, a) in tab})