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)
