def test_ignore_texts(self): """ TotalLocationEvents() can get the total activity at cell level excluding texts. """ te = TotalLocationEvents("2016-01-01", "2016-01-04", level="versioned-site", table="events.calls") df = te.get_dataframe() self.assertIs(type(df), DataFrame) # Test one of the values df.date = df.date.astype(str) val = list(df[(df.date == "2016-01-01") & (df.site_id == "0xqNDj") & (df.hour == 3)].total)[0] self.assertEqual(val, 3)
def test_only_incoming(self): """ TotalLocationEvents() can get activity, ignoring outgoing calls. """ te = TotalLocationEvents("2016-01-01", "2016-01-04", level="versioned-site", direction="in") df = te.get_dataframe() self.assertIs(type(df), DataFrame) # Test one of the values df.date = df.date.astype(str) val = list(df[(df.date == "2016-01-01") & (df.site_id == "6qpN0p") & (df.hour == 0)].total)[0] self.assertEqual(val, 2)
def test_events_at_cell_level(self): """ TotalLocationEvents() returns data at the level of the cell. """ te = TotalLocationEvents("2016-01-01", "2016-01-04", level="versioned-site") df = te.get_dataframe() self.assertIs(type(df), DataFrame) # Test one of the values df.date = df.date.astype(str) val = list(df[(df.date == "2016-01-03") & (df.site_id == "zArRjg") & (df.hour == 17)].total)[0] self.assertEqual(val, 3)
def test_events_daily(self): """ TotalLocationEvents() can get activity on a daily level. """ te = TotalLocationEvents("2016-01-01", "2016-01-04", level="versioned-site", interval="day") df = te.get_dataframe() self.assertIs(type(df), DataFrame) self.assertEqual(list(df.columns), ["site_id", "version", "lon", "lat", "date", "total"]) # Test one of the values df.date = df.date.astype(str) val = list(df[(df.date == "2016-01-03") & (df.site_id == "B8OaG5")].total)[0] self.assertEqual(val, 95)
def test_events_min(self): """ TotalLocationEvents() can get events on a min-by-min basis. """ te = TotalLocationEvents("2016-01-01", "2016-01-04", level="versioned-site", interval="min") df = te.get_dataframe() self.assertIs(type(df), DataFrame) self.assertEqual( list(df.columns), [ "site_id", "version", "lon", "lat", "date", "hour", "min", "total" ], ) # Test one of the values df.date = df.date.astype(str) val = list(df[(df.date == "2016-01-03") & (df.site_id == "zdNQx2") & (df.hour == 15) & (df["min"] == 20)].total)[0] self.assertEqual(val, 1)