def test_sparse(par): """Sparse matrix representation""" diag = np.arange(par["nx"]) + par["imag"] * np.arange(par["nx"]) D = np.diag(diag) Dop = Diagonal(diag, dtype=par["dtype"]) S = Dop.tosparse() assert_array_equal(S.A, D)
def test_sparse(par): """Sparse matrix representation """ diag = np.arange(par['nx']) +\ par['imag'] * np.arange(par['nx']) D = np.diag(diag) Dop = Diagonal(diag, dtype=par['dtype']) S = Dop.tosparse() assert_array_equal(S.A, D)