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()))