def test_square_indices_scalar(self):
for case, expected in zip(self.test_cases, zip(*self.expected)):
actual = phdlib.square_indices(self.n, case)
self.assertEqual(actual, expected)
self.assertIsInstance(actual[0], np.integer)
self.assertIsInstance(actual[1], np.integer)
def test_square_indices_list(self):
actual = phdlib.square_indices(self.n, self.test_cases)
np.testing.assert_equal(actual, self.expected)
