def test_if_program_works_with_correct_input(self):
     result = magic_square([[23, 28, 21], [22, 24, 26], [27, 20, 25]])
     result2 = magic_square([[7, 12, 1, 14], [2, 13, 8, 11],
                             [16, 3, 10, 5], [9, 6, 15, 4]])
     result3 = magic_square([[4, 9, 2], [3, 5, 7], [8, 1, 6]])
     self.assertTrue(result)
     self.assertTrue(result2)
     self.assertTrue(result3)
 def test_square_row_numbers_sum(self):
     result = magic_square([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
     self.assertFalse(result)
 def test_other_diagonal_of_matrix_numbers_sum(self):
     result = magic_square([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
     self.assertFalse(result)