def test_h(self):
"""Test compute returns the correct water level.
"""
self.bucket.indraft_fraction = 0.04
previous_volume = self.bucket.init_water_level * self.bucket.surface
evaporation = 20
precipitation = 5
seepage = 10
expected_storage = 413025.34
result_tuple = compute(self.bucket, previous_volume, evaporation,
precipitation, seepage)
self.assertAlmostEqual(expected_storage, result_tuple[0])
def test_j(self):
"""Test compute returns the correct net drainage.
"""
self.bucket.equi_water_level = 0.50
self.bucket.indraft_fraction = 0.04
previous_volume = self.bucket.init_water_level * self.bucket.surface
evaporation = 20
precipitation = 5
seepage = 10
expected_drainage = -41302.53
result_tuple = compute(self.bucket, previous_volume, evaporation,
precipitation, seepage)
self.assertAlmostEqual(expected_drainage, result_tuple[2], 2)
def test_i(self):
"""Test compute returns the correct flow off.
"""
self.bucket.equi_water_level = 0.50
self.bucket.indraft_fraction = 0.04
previous_volume = self.bucket.init_water_level * self.bucket.surface
evaporation = 20
precipitation = 5
seepage = 10
expected_flow_off = -655677.73
result_tuple = compute(self.bucket, previous_volume, evaporation,
precipitation, seepage)
self.assertAlmostEqual(expected_flow_off, result_tuple[1], 2)
