def test_location_introversion_column_names(exemplar_level_param):
""" Test that column_names property matches head(0)"""
if exemplar_level_param["level"] == "versioned-site":
pytest.skip(
'The level "versioned-site" is currently not supported in the `LocationIntroversion()` class.'
)
li = LocationIntroversion("2016-01-01", "2016-01-07", **exemplar_level_param)
assert li.head(0).columns.tolist() == li.column_names
def test_location_introversion_column_names(exemplar_spatial_unit_param):
"""Test that column_names property matches head(0)"""
li = LocationIntroversion(
"2016-01-01", "2016-01-07", spatial_unit=exemplar_spatial_unit_param
)
assert li.head(0).columns.tolist() == li.column_names