def test_evacuated_population_needs(self):
"""Test evacuated_population_needs function."""
water = ResourceParameter()
water.name = 'Water'
water.unit.name = 'litre'
water.unit.abbreviation = 'l'
water.unit.plural = 'litres'
water.frequency = 'weekly'
water.maximum_allowed_value = 10
water.minimum_allowed_value = 0
water.value = 5
rice = ResourceParameter()
rice.name = 'Rice'
rice.unit.name = 'kilogram'
rice.unit.abbreviation = 'kg'
rice.unit.plural = 'kilograms'
rice.frequency = 'daily'
rice.maximum_allowed_value = 1
rice.minimum_allowed_value = 0
rice.value = 0.5
total_needs = evacuated_population_needs(
10,
[water.serialize(), rice.serialize()]
)
self.assertEqual(total_needs['weekly'][0]['name'], 'Water')
self.assertEqual(total_needs['weekly'][0]['amount'], 50)
self.assertEqual(total_needs['weekly'][0]['table name'], 'Water [l]')
self.assertEqual(total_needs['daily'][0]['name'], 'Rice')
self.assertEqual(total_needs['daily'][0]['amount'], 5)
self.assertEqual(total_needs['daily'][0]['table name'], 'Rice [kg]')
def test_evacuated_population_needs(self):
"""Test evacuated_population_needs function."""
water = ResourceParameter()
water.name = 'Water'
water.unit.name = 'litre'
water.unit.abbreviation = 'l'
water.unit.plural = 'litres'
water.frequency = 'weekly'
water.maximum_allowed_value = 10
water.minimum_allowed_value = 0
water.value = 5
rice = ResourceParameter()
rice.name = 'Rice'
rice.unit.name = 'kilogram'
rice.unit.abbreviation = 'kg'
rice.unit.plural = 'kilograms'
rice.frequency = 'daily'
rice.maximum_allowed_value = 1
rice.minimum_allowed_value = 0
rice.value = 0.5
total_needs = evacuated_population_needs(
10, [water.serialize(), rice.serialize()])
self.assertEqual(total_needs['weekly'][0]['name'], 'Water')
self.assertEqual(total_needs['weekly'][0]['amount'], 50)
self.assertEqual(total_needs['weekly'][0]['table name'], 'Water [l]')
self.assertEqual(total_needs['daily'][0]['name'], 'Rice')
self.assertEqual(total_needs['daily'][0]['amount'], 5)
self.assertEqual(total_needs['daily'][0]['table name'], 'Rice [kg]')
