def test_1darray_sparse(self, srho_dot_ls):
rho, ham, gamma, ls, rhod = srho_dot_ls
foo = lindblad_eq(ham, ls, gamma)
rhod2 = foo(None, rho.A.reshape(-1)).reshape(3, 3)
assert_allclose(rhod, rhod2)
def test_matrix(self, rho_dot_ls):
rho, ham, gamma, ls, rhod = rho_dot_ls
foo = lindblad_eq(ham, ls, gamma)
rhod2 = foo(None, rho).reshape(3, 3)
assert_allclose(rhod, rhod2)