def test_redistribution_weights(tmpdir, tariff_data, short_form, hce):
"""Verify the shape of structure of the redistribution weights is correct
for one country."""
prep = TariffPrep(tariff_data,
working_dir_path=tmpdir.strpath,
short_form=short_form,
options={
'hce': hce,
'free_text': True,
'hiv': True,
'malaria': True,
'chinese': False
},
country='USA')
undetermined_weights = prep._get_undetermined_matrix()
cause_list = set(prep.data_module.CAUSES[cause]
for _, cause in prep.data_module.CAUSE_REDUCTION.items())
for key, weights in undetermined_weights.items():
age, sex = key
assert sex in [1, 2, 3]
if prep.AGE_GROUP == 'adult':
assert age in range(10, 81, 5) + [99]
elif prep.AGE_GROUP == 'child':
assert age in [0, 1, 5, 10, 99]
elif prep.AGE_GROUP == 'neonate':
assert age in [0, 7, 99]
assert not cause_list.symmetric_difference(weights.keys())
assert np.allclose(sum(weights.values()), 1)
def test_redistribution_weights_no_country(tmpdir, tariff_data):
prep = TariffPrep(tariff_data,
working_dir_path=tmpdir.strpath,
short_form=True,
options={
'hce': True,
'free_text': True,
'hiv': True,
'malaria': True,
'chinese': False
},
country=None)
undetermined_weights = prep._get_undetermined_matrix()
assert undetermined_weights == {}
def test_redistribution_weights_for_countries(tmpdir, tariff_data, country):
"""Any valid country from the countries list should be present and contain
the default key used for invalid age-sex observations."""
prep = TariffPrep(tariff_data,
working_dir_path=tmpdir.strpath,
short_form=True,
options={
'hce': True,
'free_text': True,
'hiv': True,
'malaria': True,
'chinese': False
},
country=country)
undetermined_weights = prep._get_undetermined_matrix()
assert isinstance(undetermined_weights, dict)
assert (99, 3) in undetermined_weights