def test_sum_of_diagonals_in_square(self): self.assertEqual(my_module.sum_of_diagonals_in_square(5), 101)
import my_module print(my_module.sum_of_diagonals_in_square(1001))