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