def test_b(self):
"""Test the case with only positive seepage on two days."""
precipitation = TimeseriesStub((self.today, 4.0),
(self.tomorrow, 6.0))
evaporation = TimeseriesStub((self.today, 2.0),
(self.tomorrow, 4.0))
seepage = TimeseriesStub((self.today, 1.0),
(self.tomorrow, 3.0))
vertical_timeseries = VerticalTimeseriesComputer()
timeseries = vertical_timeseries.compute(self.surface,
self.crop_evaporation_factor,
precipitation,
evaporation,
seepage)
expected_timeseries = {"precipitation":TimeseriesStub((self.today, .4),
(self.tomorrow, .6)),
"evaporation":TimeseriesStub((self.today, -.2),
(self.tomorrow, -.4)),
"seepage":TimeseriesStub((self.today, .1),
(self.tomorrow, .3)),
"infiltration":TimeseriesStub((self.today, 0), (self.tomorrow, 0))}
self.assertEqual(list(expected_timeseries["precipitation"].events()), list(timeseries["precipitation"].events()))
self.assertEqual(list(expected_timeseries["evaporation"].events()), list(timeseries["evaporation"].events()))
self.assertEqual(list(expected_timeseries["seepage"].events()), list(timeseries["seepage"].events()))
self.assertEqual(list(expected_timeseries["infiltration"].events()), list(timeseries["infiltration"].events()))
def test_aa(self):
"""Test the case with only positive seepage on a single day.
The crop evaporation factor is 0.5.
"""
precipitation = TimeseriesStub((self.today, 4.0))
evaporation = TimeseriesStub((self.today, 2.0))
seepage = TimeseriesStub((self.today, 1.0))
vertical_timeseries = VerticalTimeseriesComputer()
timeseries = vertical_timeseries.compute(self.surface,
0.5,
precipitation,
evaporation,
seepage)
expected_timeseries = {"precipitation":TimeseriesStub((self.today, .4)),
"evaporation":TimeseriesStub((self.today, -.1)),
"seepage":TimeseriesStub((self.today, .1)),
"infiltration":TimeseriesStub((self.today, 0))}
self.assertEqual(list(expected_timeseries["precipitation"].events()), list(timeseries["precipitation"].events()))
self.assertEqual(list(expected_timeseries["evaporation"].events()), list(timeseries["evaporation"].events()))
self.assertEqual(list(expected_timeseries["seepage"].events()), list(timeseries["seepage"].events()))
self.assertEqual(list(expected_timeseries["infiltration"].events()), list(timeseries["infiltration"].events()))
