def test_dop_1darray_sparse(self, srho_dot):
rho, ham, rhod = srho_dot
foo = schrodinger_eq_dop(ham)
rhod2 = foo(None, rho.A.reshape(-1)).reshape(3, 3)
assert_allclose(rhod, rhod2, atol=1e-12)
def test_dop_1darray(self, rho_dot):
rho, ham, rhod = rho_dot
foo = schrodinger_eq_dop(ham)
rhod2 = foo(None, rho.A.reshape(-1)).reshape(3, 3)
assert_allclose(rhod, rhod2)