def test_variable(self, model):
""" Test that variable updating works. """
p1 = AnnualHarmonicSeriesParameter(model,
0.5, [0.25], [np.pi / 4],
is_variable=True)
assert p1.double_size == 3
assert p1.integer_size == 0
new_var = np.array([0.6, 0.1, np.pi / 2])
p1.set_double_variables(new_var)
np.testing.assert_allclose(p1.get_double_variables(), new_var)
with pytest.raises(NotImplementedError):
p1.set_integer_variables(np.arange(3, dtype=np.int32))
with pytest.raises(NotImplementedError):
p1.get_integer_variables()
si = ScenarioIndex(0, np.array([0], dtype=np.int32))
for ts in model.timestepper:
doy = (ts.datetime.dayofyear - 1) / 365
np.testing.assert_allclose(
p1.value(ts, si),
0.6 + 0.1 * np.cos(doy * 2 * np.pi + np.pi / 2))
def test_double_harmonic(self, model):
p1 = AnnualHarmonicSeriesParameter(model, 0.5, [0.25, 0.3], [np.pi/4, np.pi/3])
si = ScenarioIndex(0, np.array([0], dtype=np.int32))
for ts in model.timestepper:
doy = (ts.datetime.dayofyear - 1) /365
expected = 0.5 + 0.25*np.cos(doy*2*np.pi + np.pi / 4) + 0.3*np.cos(doy*4*np.pi + np.pi/3)
np.testing.assert_allclose(p1.value(ts, si), expected)
def test_single_harmonic(self, model):
p1 = AnnualHarmonicSeriesParameter(0.5, [0.25], [np.pi / 4])
si = ScenarioIndex(0, np.array([0], dtype=np.int32))
for ts in model.timestepper:
doy = (ts.datetime.dayofyear - 1) / 365
np.testing.assert_allclose(
p1.value(ts, si),
0.5 + 0.25 * np.cos(doy * 2 * np.pi + np.pi / 4))