def test_column_names(exemplar_spatial_unit_param, ignore_nulls): """Test that column_names property matches head(0) for SubscriberLocations""" sl = SubscriberLocations( "2016-01-01", "2016-01-04", spatial_unit=exemplar_spatial_unit_param, ignore_nulls=ignore_nulls, ) assert sl.head(0).columns.tolist() == sl.column_names
def __init__( self, start, stop, *, spatial_unit: AnySpatialUnit = make_spatial_unit("cell"), subscriber_identifier="msisdn", hours: Optional[Tuple[int, int]] = None, subscriber_subset=None, tables="all", ignore_nulls=True, ): self.subscriber_locations = SubscriberLocations( start=start, stop=stop, spatial_unit=spatial_unit, table=tables, hours=hours, subscriber_identifier=subscriber_identifier, subscriber_subset=subscriber_subset, ignore_nulls=ignore_nulls, ) super().__init__()
def test_can_get_pcods(get_dataframe): """ SubscriberLocations() can make queries at the p-code level. """ subscriber_pcod = SubscriberLocations( "2016-01-01 13:30:30", "2016-01-02 16:25:00", spatial_unit=make_spatial_unit("polygon", region_id_column_name="admin3pcod", geom_table="geography.admin3"), ) df = get_dataframe(subscriber_pcod) assert df.admin3pcod[0].startswith("524")
def _flowmachine_query_obj(self): """ Return the underlying flowmachine object. Returns ------- Query """ return RedactedTripsODMatrix(trips=TripsODMatrix( SubscriberLocations( self.start_date, self.end_date, spatial_unit=self.aggregation_unit, table=self.event_types, subscriber_subset=self.subscriber_subset, )))