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))