diff --git a/exercises/saddle-points/saddle_points_test.py b/exercises/saddle-points/saddle_points_test.py index c614f21469f..1f1cf846898 100644 --- a/exercises/saddle-points/saddle_points_test.py +++ b/exercises/saddle-points/saddle_points_test.py @@ -10,7 +10,7 @@ from saddle_points import saddle_points -# Tests adapted from `problem-specifications//canonical-data.json` @ v1.1.0 +# Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.0 class SaddlePointsTest(unittest.TestCase): def test_identify_single_saddle_point(self): @@ -24,11 +24,16 @@ def test_identify_lack_of_saddle_points_when_there_are_none(self): matrix = [[1, 2, 3], [3, 1, 2], [2, 3, 1]] self.assertEqual(saddle_points(matrix), set()) - def test_identify_multiple_saddle_points(self): + def test_identify_multiple_saddle_points_in_column(self): matrix = [[4, 5, 4], [3, 5, 5], [1, 5, 4]] expected = set([(0, 1), (1, 1), (2, 1)]) self.assertEqual(saddle_points(matrix), expected) + def test_identify_multiple_saddle_points_in_row(self): + matrix = [[6, 7, 8], [5, 5, 5], [7, 5, 6]] + expected = set([(1, 0), (1, 1), (1, 2)]) + self.assertEqual(saddle_points(matrix), expected) + def test_identify_saddle_point_in_bottom_right_corner(self): matrix = [[8, 7, 9], [6, 7, 6], [3, 2, 5]] expected = set([(2, 2)])