def test_equals_2(self):
     self.assertEqual(
         main_diagonal_product([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), 45)
 def test_equals_2(self):
     self.assertEqual(main_diagonal_product(
         [[1, 2, 3], [4, 5, 6], [7, 8, 9]]), 45
     )
 def test_equals(self):
     self.assertEqual(main_diagonal_product([[1, 0], [0, 1]]), 1)
 def test_equals(self):
     self.assertEqual(main_diagonal_product([[1, 0], [0, 1]]), 1)