def test_given_only_one_ccaa_when_calculate_global_incidence_then_same_value_returned(
self):
ia = 7
result = calculate_global_incidence(
{"Castilla y León": {
Measurement.ACCUMULATED_INCIDENCE: ia
}}, Measurement.ACCUMULATED_INCIDENCE)
# Round because the operation is returning 6.999999
self.assertEqual(ia, round(result, 0))
def test_given_two_ccaas_when_calculate_global_incidence_then_same_value_returned(
self):
ia1 = 7
ia2 = 6
ccaa1 = "Castilla y León"
ccaa2 = "Castilla La Mancha"
result = calculate_global_incidence(
{
ccaa1: {
Measurement.ACCUMULATED_INCIDENCE: ia1
},
ccaa2: {
Measurement.ACCUMULATED_INCIDENCE: ia2
}
}, Measurement.ACCUMULATED_INCIDENCE)
population_ccaa1 = CCAA_POPULATION[ccaa1]
population_ccaa2 = CCAA_POPULATION[ccaa2]
total_cases = ia1 * population_ccaa1 / 100000 + ia2 * population_ccaa2 / 100000
total_population = population_ccaa1 + population_ccaa2
expected_result = total_cases / total_population * 100000
self.assertEqual(expected_result, result)
def test_given_no_data_when_calculate_global_incidence_then_zero_returned(
self):
self.assertEqual(
0, calculate_global_incidence({},
Measurement.ACCUMULATED_INCIDENCE))