def test_travel_displacement():
from text_models.place import Travel, CP
travel = Travel(DATE, window=3)
assert len(travel._days) == 3
output = travel.displacement()
for v in output.values():
assert len(v) == 3
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_outward():
from text_models.place import Travel
from datetime import datetime
travel = Travel(DATE, window=4)
matrix = travel.outward(travel.country)
outward = matrix[-1]["MX"]
assert outward
assert "MX" not in outward
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)
def test_travel_dates():
from text_models.place import Travel, CP
from datetime import datetime
travel = Travel(datetime(year=2020, month=4, day=4), window=4)
print(travel.dates[-1].day, 4)
assert travel.dates[-1].day == 4
print(travel.dates[0].day, 1)
assert travel.dates[0].day == 1
def test_travel_init():
from text_models.place import Travel, CP
from datetime import datetime
travel = Travel(datetime(year=2015, month=12, day=18), window=4)
print(len(travel.dates))
assert len(travel.dates) == 3
def test_travel():
from text_models.place import Travel, CP
travel = Travel(DATE, window=3)
assert len(travel._days) == 3
print(travel.num_users)