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)