def test_join_to_location_column_names(exemplar_level_param):
""" Test that JoinToLocation's column_names property is accurate."""
if "cell" == exemplar_level_param["level"]:
pytest.skip("Cell level not valid for JoinToLocation"
) # cell level not valid for JoinToLocation
table = subscriber_locations("2016-01-05", "2016-01-07", level="cell")
joined = JoinToLocation(table, **exemplar_level_param)
assert joined.head(0).columns.tolist() == joined.column_names
def test_join_to_location_column_names(exemplar_spatial_unit_param):
""" Test that JoinToLocation's column_names property is accurate."""
if not exemplar_spatial_unit_param.has_geography:
pytest.skip("JoinToLocation does not accept CellSpatialUnit objects")
table = SubscriberLocations("2016-01-05",
"2016-01-07",
spatial_unit=make_spatial_unit("cell"))
joined = JoinToLocation(table, spatial_unit=exemplar_spatial_unit_param)
assert joined.head(0).columns.tolist() == joined.column_names
