def test_failures():
with pytest.raises(rtd.RTDInputError):
rtd.AD_oo(tau=-1, peclet=1, dt=DT, time_end=TIME_END)
with pytest.raises(rtd.RTDInputError):
rtd.AD_oo(tau=1, peclet=-1, dt=DT, time_end=TIME_END)
with pytest.raises(rtd.RTDInputError):
rtd.AD_oo(tau=1, peclet=1, dt=-1, time_end=TIME_END)
with pytest.raises(rtd.RTDInputError):
rtd.AD_oo(tau=1, peclet=1, dt=DT, time_end=-1)
def test1(tau, peclet):
a = rtd.AD_oo(tau=tau, peclet=peclet, dt=DT, time_end=TIME_END)
assert(np.isclose(a.integral(), 1, rtol=rtol, atol=atol))
assert(np.isclose(a.mrt(), analytic_mrt(tau, peclet),
rtol=rtol, atol=atol))
assert(np.isclose(a.tau_oo, analytic_mrt(tau, peclet),
rtol=rtol, atol=atol))
assert(np.isclose(a.sigma(), analytic_sigma2(tau, peclet),
rtol=rtol, atol=atol))
def test_low_high_tau(tau, peclet):
time_end = 8000
dt = .005
a = rtd.AD_oo(tau=tau, peclet=peclet, dt=dt, time_end=time_end)
assert(np.isclose(a.integral(), 1, rtol=rtol, atol=atol))
assert(np.isclose(a.mrt(), analytic_mrt(tau, peclet),
rtol=rtol, atol=atol))
assert(np.isclose(a.tau_oo, analytic_mrt(tau, peclet),
rtol=rtol, atol=atol))
assert(np.isclose(a.sigma(), analytic_sigma2(tau, peclet),
rtol=rtol, atol=atol))
def test_AD_equivalence(peclet):
# closed-closed and open-open are approximately equal at high peclet
# starting around Pe=100
time_end = 10
dt = .001
tau = 1
nx = 300
a = rtd.AD_cc(tau=tau, peclet=peclet, dt=dt, time_end=time_end, nx=nx)
b = rtd.AD_oo(tau=tau, peclet=peclet, dt=dt, time_end=time_end)
assert(np.isclose(a.integral(), b.integral(), rtol=rtol, atol=atol))
assert(np.isclose(a.mrt(), b.mrt(), rtol=1e-2, atol=atol))
assert(np.isclose(a.sigma(), b.sigma(), rtol=5e-2, atol=atol))
import matplotlib.pyplot as plt
import rtdpy
for pe in [10, 100]:
a = rtdpy.AD_oo(tau=1, peclet=pe, dt=.01, time_end=3)
plt.plot(a.time, a.exitage, label=f"peclet={pe}")
plt.xlabel('Time')
plt.ylabel('Exit Age Function')
plt.legend()
plt.show()
def test_repr():
a = rtd.AD_oo(tau=1, peclet=100, dt=DT, time_end=TIME_END)
b = eval("rtd."+repr(a))
assert np.isclose(a.integral(), b.integral(), rtol=rtol, atol=atol)
assert np.isclose(a.mrt(), b.mrt(), rtol=rtol, atol=atol)
assert np.isclose(a.sigma(), b.sigma(), rtol=rtol, atol=atol)