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(program.riverSizes(testInput)), expected)
def test_case_1(self):
# [1, 0, 0, 1, 0],
# [1, 0, 1, 0, 0],
# [0, 0, 1, 0, 1],
# [1, 0, 1, 0, 1],
# [1, 0, 1, 1, 0]
testInput = [[1, 0, 0, 1, 0], [1, 0, 1, 0, 0], [0, 0, 1, 0, 1], [1, 0, 1, 0, 1], [1, 0, 1, 1, 0]]
expected = [1, 2, 2, 2, 5]
self.assertEqual(sorted(program.riverSizes(testInput)), expected)
print("pass")
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(program.riverSizes(testInput)), expected)
def test_case_4(self):
testInput = [[1, 0, 0, 1], [1, 0, 1, 0], [0, 0, 1, 0], [1, 0, 1, 0]]
expected = [1, 1, 2, 3]
self.assertEqual(sorted(program.riverSizes(testInput)), expected)
def test_case_2(self):
testInput = [[1]]
expected = [1]
self.assertEqual(sorted(program.riverSizes(testInput)), expected)
3
4
5
class TestProgram(unittest.TestCase):
6
7
def test_case_1(self):
8
testInput = [[0]]
9
expected = []
10
self.assertEqual(sorted(program.riverSizes(testInput)), expected)
11
12
def test_case_2(self):
13
testInput = [[1]]
14
expected = [1]
15
self.assertEqual(sorted(program.riverSizes(testInput)), expected)
16
17
def test_case_3(self):
18