def test_ket_1darray_sparse(self, spsi_dot):
psi, ham, psid = spsi_dot
foo = schrodinger_eq_ket(ham)
psid2 = foo(None, psi.A.reshape(-1)).reshape(-1, 1)
assert_allclose(psid, psid2)
def test_ket_matrix_sparse(self, spsi_dot):
psi, ham, psid = spsi_dot
foo = schrodinger_eq_ket(ham)
psid2 = foo(None, psi)
assert_allclose(psid, psid2)
