def test_partition():
tests = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for test in tests:
f = partition(test)
while True:
try:
l = next(f)
except StopIteration:
break
tests_k = [8, 10]
for test in tests_k:
for k in range(8):
f = partition(test, k)
while True:
try:
l = next(f)
assert len(l) == k
except StopIteration:
break
def test_partition():
tests = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for test in tests:
f = partition(test)
while True:
try:
l = next(f)
except StopIteration:
break
tests_k = [8, 10]
for test in tests_k:
for k in range(8):
f = partition(test, k)
while True:
try:
l = next(f)
assert len(l) == k
except StopIteration:
break
def test_diop_partition():
for n in [8, 10]:
for k in range(1, 8):
for p in partition(n, k):
assert len(p) == k
assert [p for p in partition(3, 5)] == []
assert [list(p) for p in partition(3, 5, 1)] == [[0, 0, 0, 0, 3], [0, 0, 0, 1, 2], [0, 0, 1, 1, 1]]
assert list(partition(0)) == [()]
assert list(partition(1, 0)) == [()]
assert [list(i) for i in partition(3)] == [[1, 1, 1], [1, 2], [3]]
def test_diop_partition():
for n in [8, 10]:
for k in range(1, 8):
for p in partition(n, k):
assert len(p) == k
assert [p for p in partition(3, 5)] == []
assert [list(p) for p in partition(3, 5, 1)] == [
[0, 0, 0, 0, 3], [0, 0, 0, 1, 2], [0, 0, 1, 1, 1]]
assert list(partition(0)) == [()]
assert list(partition(1, 0)) == [()]
assert [list(i) for i in partition(3)] == [[1, 1, 1], [1, 2], [3]]
