def test_invalid_list_type_display_names(self):
field_mappings = FieldMappings(client=ES_TEST_CLIENT,
index_pattern=FLIGHTS_INDEX_NAME)
# not a list like object
with pytest.raises(ValueError):
field_mappings.display_names = 12.0
# tuple is list like
field_mappings.display_names = ("Cancelled", "DestWeather")
expected = ["Cancelled", "DestWeather"]
assert expected == field_mappings.display_names
def test_set_display_names(self):
expected = [
"Cancelled",
"timestamp",
"DestWeather",
"DistanceKilometers",
"AvgTicketPrice",
]
field_mappings = FieldMappings(client=ES_TEST_CLIENT,
index_pattern=FLIGHTS_INDEX_NAME)
field_mappings.display_names = expected
assert expected == field_mappings.display_names
# now set again
new_expected = ["AvgTicketPrice", "timestamp"]
field_mappings.display_names = new_expected
assert new_expected == field_mappings.display_names
def test_not_found_display_names(self):
not_found = [
"Cancelled",
"timestamp",
"DestWeather",
"unknown",
"DistanceKilometers",
"AvgTicketPrice",
]
field_mappings = FieldMappings(client=ES_TEST_CLIENT,
index_pattern=FLIGHTS_INDEX_NAME)
with pytest.raises(KeyError):
field_mappings.display_names = not_found
expected = self.pd_flights().columns.to_list()
assert expected == field_mappings.display_names
