@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_ndim={}_n={}'.format(ndim, n), 'ndim': ndim, 'n': n} for ndim in [0, 1, 4] for n in [0, 1, 7])))
def testDiagIndices(self, ndim, n):
    onp.testing.assert_equal(onp.diag_indices(n, ndim), jnp.diag_indices(n, ndim))
@parameterized.named_parameters(jtu.cases_from_list(({'testcase_name': '_ndim={}_n={}'.format(ndim, n), 'ndim': ndim, 'n': n} for ndim in [0, 1, 4] for n in [0, 1, 7])))
def testDiagIndices(self, ndim, n):
    onp.testing.assert_equal(onp.diag_indices(n, ndim), jnp.diag_indices(n, ndim))