def test_solution_1_matrix_size_3(self):
in_matrix = [[x for x in range(0, 3)] for _ in range(3)]
out_matrix = [
[0, 2, 6],
[0, 9, 8],
[0, 6, 10]
]
snail = Snail()
self.assertEqual(out_matrix, snail.solution_1(in_matrix), f'in_matrix:{in_matrix}')
def test_solution_1_matrix_size_2(self):
in_matrix = [
[0, 1],
[0, 1]
]
out_matrix = [
[0, 2],
[0, 3]
]
snail = Snail()
self.assertEqual(out_matrix, snail.solution_1(in_matrix))
def test_solution_1_matrix_size_1(self):
in_matrix = [[0]]
snail = Snail()
self.assertEqual(in_matrix, snail.solution_1(in_matrix))
