def __init__(self, w_avg, N_STP, r_dry, mass_of_dry_air):
self.q0 = const.eps / (self.p0 / self.RH0 / phys.pvs(self.T0) - 1)
self.w_avg = w_avg
self.r_dry = r_dry
self.N_STP = N_STP
self.n_in_dv = N_STP / const.rho_STP * mass_of_dry_air
self.mass_of_dry_air = mass_of_dry_air
def test_initialisation(setup_idx):
# Arrange
setup = setups[setup_idx]
pv0 = setup.p0 / (1 + const.eps / setup.q0)
pd0 = setup.p0 - pv0
rhod0 = pd0 / const.Rd / setup.T0
thd0 = phys.th_std(pd0, setup.T0)
# Act
simulation = Simulation(setup)
# Assert
env = simulation.particles.environment
np.testing.assert_approx_equal(env['T'][:], setup.T0)
np.testing.assert_approx_equal(env['RH'][:], pv0 / phys.pvs(setup.T0))
np.testing.assert_approx_equal(env['p'][:], setup.p0)
np.testing.assert_approx_equal(env['qv'][:], setup.q0)
np.testing.assert_approx_equal(env['rhod'][:], rhod0)
np.testing.assert_approx_equal(env['thd'][:], thd0)