Exemplo n.º 1
0
 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)
Exemplo n.º 2
0
 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)
Exemplo n.º 3
0
 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)
Exemplo n.º 4
0
 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)
Exemplo n.º 5
0
 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)
Exemplo n.º 6
0
 def test_case_2(self):
     testInput = [[1]]
     expected = [1]
     self.assertEqual(sorted(riverSizes(testInput)), expected)