def test_create_stat_desc_dic(sample_stat_meta_response):
desc_dict = StatisticsGraphQlMetaDataProvider._create_stat_desc_dic(
sample_stat_meta_response)
assert desc_dict["AENW01"][
0] == "Description 1", "first dict entry is wrong"
assert (desc_dict["AENW02"][0] == "NO DESCRIPTION FOUND"
), "second dict entry is wrong"
def test_get_query_specific_enum_meta():
field_type_list = [
("id", "String"),
("name", "String"),
("WAHL09", "WAHL09"),
("PART04", "PART04"),
("year", "Int"),
("value", "Float"),
]
query_stat_meta = StatisticsGraphQlMetaDataProvider().get_query_enum_meta(
field_type_list)
expected_stat_meta = {
"PART04":
dict([
("AFD", "AfD"),
("B90_GRUENE", "GRÜNE"),
("CDU", "CDU/CSU"),
("DIELINKE", "DIE LINKE"),
("FDP", "FDP"),
("SONSTIGE", "Sonstige Parteien"),
("SPD", "SPD"),
("GESAMT", "Gesamt"),
])
}
assert query_stat_meta == expected_stat_meta
def test_extract_stat_desc(stat_description):
extracted_text = StatisticsGraphQlMetaDataProvider._extract_main_description(
stat_description[0]
)
assert (
extracted_text == stat_description[1]
), "extracted text does not match bold part"
def test_filter_stat_metatdata(sample_stat_meta_response):
processed_stat_meta = StatisticsGraphQlMetaDataProvider._process_stat_meta_data(
sample_stat_meta_response)
assert all(
all(("name" in stat, "description" in stat, "args" in stat))
for stat in processed_stat_meta)
assert all(
any(arg["name"] == "statistics" for arg in field["args"])
for field in processed_stat_meta)
assert len(processed_stat_meta) == 2, "too many fields filtered out"
def test_get_query_specific_stat_meta():
field_type_list = [
("id", "String"),
("name", "String"),
("WAHL09", "WAHL09"),
("PART04", "PART04"),
("year", "Int"),
("value", "Float"),
]
query_stat_meta = StatisticsGraphQlMetaDataProvider().get_query_stat_meta(
field_type_list)
# "Gültige Zweitstimmen" uninformative due to API changes
expected_stat_meta = {"WAHL09": "WAHL09"}
assert query_stat_meta == expected_stat_meta
def test_get_query_specific_unit_meta():
field_type_list = [
("id", "String"),
("name", "String"),
("WAHL09", "WAHL09"),
("PART04", "PART04"),
("year", "Int"),
("value", "Float"),
]
query_stat_meta = StatisticsGraphQlMetaDataProvider().get_query_unit_meta(
field_type_list)
expected_stat_meta = {
"WAHL09":
"StatisticsGraphQlMetaDataProvider does not provide unit information."
}
assert query_stat_meta == expected_stat_meta