def test_case_11(self):
testInput = [
[1, 1, 0, 0, 0, 0, 1, 1],
[1, 0, 1, 1, 1, 1, 0, 1],
[0, 1, 1, 0, 0, 0, 1, 1],
]
expected = [3, 5, 6]
self.assertEqual(sorted(riverSizes(testInput)), expected)
def test_case_6(self):
testInput = [
[1, 0, 0, 1, 0, 1, 0, 0, 1, 1, 1, 0],
[1, 0, 1, 0, 0, 1, 1, 1, 1, 0, 1, 0],
[0, 0, 1, 0, 1, 1, 0, 1, 0, 1, 1, 1],
[1, 0, 1, 0, 1, 1, 0, 0, 0, 1, 0, 0],
[1, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0, 1],
]
expected = [1, 1, 2, 2, 5, 21]
self.assertEqual(sorted(riverSizes(testInput)), expected)
def test_case_10(self):
testInput = [
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
[1, 1, 1, 1, 1, 1, 1],
]
expected = [49]
self.assertEqual(sorted(riverSizes(testInput)), expected)
def test_case_9(self):
testInput = [
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0],
]
expected = []
self.assertEqual(sorted(riverSizes(testInput)), expected)
def test_case_3(self):
testInput = [[1, 1, 1, 0, 1, 1, 0, 0, 0, 1, 0]]
expected = [1, 2, 3]
self.assertEqual(sorted(riverSizes(testInput)), expected)
def test_case_2(self):
testInput = [[1]]
expected = [1]
self.assertEqual(sorted(riverSizes(testInput)), expected)
