def test_travel_inside_mobility():
from text_models.place import Travel
from datetime import datetime
travel = Travel(DATE, window=4)
dis = travel.displacement(travel.country)
inside = travel.inside_mobility(travel.country)
assert sum(dis["MX"]) > sum(inside["MX"])
def test_travel_percentage_by_weekday2():
from text_models.place import Travel
import numpy as np
travel = Travel(DATE, window=21)
inside = travel.inside_mobility(travel.bounding_box.city)
baseline = travel.weekday_percentage(inside)
print(len(baseline))
assert len(baseline) < 14347
def test_travel_weekday():
from text_models.place import Travel
travel = Travel(DATE, window=21)
inside = travel.inside_mobility(travel.country)
baseline = travel.group_by_weekday(inside)
for wk in range(7):
_ = baseline["MX"][wk]
print(_)
assert sum(_) > 0
def test_travel_percentage_by_weekday():
from text_models.place import Travel
import numpy as np
travel = Travel(DATE, window=21)
func = lambda x: x[:2] if x[:2] in ["MX", "US"] else None
inside = travel.inside_mobility(func)
baseline = travel.weekday_percentage(inside)
print(len(inside), inside["MX"])
output = travel.transform(inside, baseline)
for k in ["MX", "US"]:
v = output[k]
print(v)
assert len([1 for x in v if x == 0]) == 7
def test_travel_weekday_percentage():
from text_models.place import Travel
travel = Travel(DATE, window=21)
inside = travel.inside_mobility(travel.country)
baseline = travel.weekday_percentage(inside)
for wk in range(7):
_ = baseline["MX"]
print(_)
assert _._data[wk] > 0
for v in baseline.values():
v.mobility_instance = travel
y = baseline["MX"].transform(inside["MX"])
assert len(y) == len(travel.dates)
def test_travel_weekday_probability():
from text_models.utils import Gaussian
from text_models.place import Travel
import numpy as np
travel = Travel(DATE, window=21)
func = lambda x: x[:2] if x[:2] == "MX" else None
inside = travel.inside_mobility(func)
baseline = travel.weekday_probability(inside)
print(baseline["MX"].data[0])
for wk in range(7):
assert isinstance(baseline["MX"].data[wk], Gaussian)
output = travel.transform(inside, baseline)
print(output["MX"])
assert np.all(output["MX"] > 0.01)