98b086f746
see https://github.com/karlicoss/HPI/issues/262 * move home to fallback/via_home.py * move via_ip to fallback * add fallback model * add stub via_ip file * add fallback_locations for via_ip * use protocol for locations * estimate_from helper, via_home estimator, all.py * via_home: add accuracy, cache history * add datasources to gpslogger/google_takeout * tz/via_location.py: update import to fallback * denylist docs/installation instructions * tz.via_location: let user customize cachew refresh time * add via_ip.estimate_location using binary search * use estimate_location in via_home.get_location * tests: add gpslogger to location config stub * tests: install tz related libs in test env * tz: add regression test for broken windows dates * vendorize bisect_left from python src doesnt have a 'key' parameter till python3.10
95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
import sys
|
|
from datetime import datetime, timedelta
|
|
from pathlib import Path
|
|
|
|
import pytest # type: ignore
|
|
import pytz # type: ignore
|
|
|
|
from my.core.error import notnone
|
|
|
|
import my.time.tz.main as TZ
|
|
import my.time.tz.via_location as LTZ
|
|
|
|
|
|
def test_iter_tzs() -> None:
|
|
ll = list(LTZ._iter_tzs())
|
|
assert len(ll) > 3
|
|
|
|
|
|
def test_past() -> None:
|
|
# should fallback to the home location provider
|
|
dt = D('20000101 12:34:45')
|
|
dt = TZ.localize(dt)
|
|
tz = dt.tzinfo
|
|
assert tz is not None
|
|
assert getattr(tz, 'zone') == 'America/New_York'
|
|
|
|
|
|
def test_future() -> None:
|
|
fut = datetime.now() + timedelta(days=100)
|
|
# shouldn't crash at least
|
|
assert TZ.localize(fut) is not None
|
|
|
|
|
|
def test_tz() -> None:
|
|
# todo hmm, the way it's implemented at the moment, never returns None?
|
|
|
|
# not present in the test data
|
|
tz = LTZ._get_tz(D('20200101 10:00:00'))
|
|
assert notnone(tz).zone == 'Europe/Sofia'
|
|
|
|
tz = LTZ._get_tz(D('20170801 11:00:00'))
|
|
assert notnone(tz).zone == 'Europe/Vienna'
|
|
|
|
tz = LTZ._get_tz(D('20170730 10:00:00'))
|
|
assert notnone(tz).zone == 'Europe/Rome'
|
|
|
|
tz = LTZ._get_tz(D('20201001 14:15:16'))
|
|
assert tz is not None
|
|
|
|
on_windows = sys.platform == 'win32'
|
|
if not on_windows:
|
|
tz = LTZ._get_tz(datetime.min)
|
|
assert tz is not None
|
|
else:
|
|
# seems this fails because windows doesnt support same date ranges
|
|
# https://stackoverflow.com/a/41400321/
|
|
with pytest.raises(OSError):
|
|
LTZ._get_tz(datetime.min)
|
|
|
|
|
|
def test_policies() -> None:
|
|
getzone = lambda dt: getattr(dt.tzinfo, 'zone')
|
|
|
|
naive = D('20170730 10:00:00')
|
|
# actual timezone at the time
|
|
assert getzone(TZ.localize(naive)) == 'Europe/Rome'
|
|
|
|
z = pytz.timezone('America/New_York')
|
|
aware = z.localize(naive)
|
|
|
|
assert getzone(TZ.localize(aware)) == 'America/New_York'
|
|
|
|
assert getzone(TZ.localize(aware, policy='convert')) == 'Europe/Rome'
|
|
|
|
|
|
with pytest.raises(RuntimeError):
|
|
assert TZ.localize(aware, policy='throw')
|
|
|
|
|
|
def D(dstr: str) -> datetime:
|
|
return datetime.strptime(dstr, '%Y%m%d %H:%M:%S')
|
|
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def prepare(tmp_path: Path):
|
|
from .shared_config import temp_config
|
|
conf = temp_config(tmp_path)
|
|
|
|
import my.core.cfg as C
|
|
with C.tmp_config() as config:
|
|
config.google = conf.google
|
|
config.time = conf.time
|
|
config.location = conf.location
|
|
yield
|