def test_normalize_else(): """ test to make sure psi is being returned un-normalized when hops.basis.eom.normalized is False """ hops = HOPS( sys_param, noise_param=noise_param, hierarchy_param=hier_param, eom_param=eom_param, integration_param=integrator_param, ) hops.initialize([2, 3]) hops.basis.eom.normalized = False norm = hops.normalize([2, 3]) known_norm = [2, 3] assert np.allclose(norm, known_norm)
def test_normalize_if(): """ test to make sure psi is being returned normalized when hops.basis.eom.normalized is True """ hops = HOPS( sys_param, noise_param=noise_param, hierarchy_param=hier_param, eom_param=eom_param, integration_param=integrator_param, ) hops.initialize([2, 3]) hops.basis.eom.normalized = True norm = hops.normalize([2, 3]) known_norm = [0.5547002, 0.83205029] assert np.allclose(norm, known_norm)