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)