def test_standardize_datetime():
"""Test ISO8601 datetime format.
"""
t0 = datetime_with_timezone(T0_DST)
t0_as_utc = datetime_with_timezone(t0, 'UTC')
datetimetuple1 = (2016, 11, 5, 23, 0, 0) # 2016-11-05 23:00:00
_, d_str1 = standardize_datetime(datetimetuple1, time_zone=US_NY_ZONE_NAME)
assert d_str1 == f"{t0_as_utc.year:4d}-{t0_as_utc.month:02d}-{t0_as_utc.day:02d}T{t0_as_utc.hour:02d}:{t0_as_utc.minute:02d}:{t0_as_utc.second:02d}.{int(t0_as_utc.microsecond/1000):03d}Z"
assert d_str1 == standardize_datetime(T0_DST)[1]
assert d_str1 == standardize_datetime(t0)[1]
def test_func_parse_dt2():
"""Test returned datetime DST or not.
"""
tt0 = datetime_with_timezone(T0_DST) # DST
assert is_dst(tt0) == True
assert is_dst(T0_DST) == True
tt0_after_four_hrs = parse_dt("after 4 hours", tt0)
assert is_dst(tt0_after_four_hrs) == False
assert (tt0_after_four_hrs - tt0).total_seconds() == 4 * 3600
assert (datetime_with_timezone(tt0_after_four_hrs, time_zone='UTC') -
datetime_with_timezone(tt0, time_zone='UTC')).total_seconds() \
== 4 * 3600
def test_func_datetime_with_timezone2():
"""Test `datetime_with_timezone()`, local zone may not fall into DST.
"""
tt0 = datetime_with_timezone(T0)
tt1 = datetime_with_timezone(T1)
assert tt0.tzinfo.zone == LOCAL_ZONE_NAME
assert tt1.tzinfo.zone == LOCAL_ZONE_NAME
# inteprete datetime at US_NY_ZONE_NAME
tt0 = datetime_with_timezone(T0, time_zone=US_NY_ZONE_NAME)
tt1 = datetime_with_timezone(T1, time_zone=US_NY_ZONE_NAME)
assert tt0.tzinfo.zone == US_NY_ZONE_NAME
assert bool(tt0.dst()) == True
assert tt1.tzinfo.zone == US_NY_ZONE_NAME
assert bool(tt1.dst()) == False
def test_func_parse_dt1():
"""Test `parse_dt()`.
"""
dt_str = "after 3 hours"
tt0 = datetime_with_timezone(T0_DST)
t0_as_utc = datetime_with_timezone(tt0, time_zone='UTC')
t0_after_utc = parse_dt(dt_str, t0_as_utc)
assert (t0_after_utc - t0_as_utc).total_seconds() / 3600 == 3.0
t0_after_local = parse_dt(dt_str, tt0)
assert (t0_after_local - tt0).total_seconds() / 3600 == 3.0
assert (t0_after_local - t0_after_utc).total_seconds() == 0.0
dt_str = "3 hours before"
t0_before_utc = parse_dt(dt_str, t0_as_utc)
assert (t0_before_utc - t0_as_utc).total_seconds() / 3600 == -3.0
t0_before_local = parse_dt(dt_str, tt0)
assert (t0_before_local - tt0).total_seconds() / 3600 == -3.0
assert (t0_before_local - t0_before_utc).total_seconds() == 0.0
def test_func_datetime_with_timezone1():
"""Test `datetime_with_timezone()`, DST.
"""
tt0 = datetime_with_timezone(T0_DST)
tt1 = datetime_with_timezone(T1_EST)
assert tt0.tzinfo.zone == US_NY_ZONE_NAME
assert bool(tt0.dst()) == True
assert bool(tt0.dst()) == is_dst(T0_DST)
assert tt1.tzinfo.zone == US_NY_ZONE_NAME
assert bool(tt1.dst()) == False
assert bool(tt1.dst()) == is_dst(T1_EST)
t0_at_utc = datetime_with_timezone(T0, time_zone='UTC')
assert (tt0 - t0_at_utc).total_seconds() / 3600 == 4.0
t1_at_utc = datetime_with_timezone(T1, time_zone='UTC')
assert (tt1 - t1_at_utc).total_seconds() / 3600 == 5.0
t0_as_utc = datetime_with_timezone(tt0, time_zone='UTC')
assert (tt0 - t0_as_utc).total_seconds() == 0.0
t1_as_utc = datetime_with_timezone(tt1, time_zone='UTC')
assert (tt1 - t1_as_utc).total_seconds() == 0.0
from datetime import datetime
import pytz
from archappl.data import is_dst
from archappl.data import parse_dt
from archappl.data import datetime_with_timezone
from archappl.data.utils import standardize_datetime
from archappl.data.utils import LOCAL_ZONE_NAME
US_NY_ZONE_NAME = 'America/New_York'
US_NY_ZONE = pytz.timezone(US_NY_ZONE_NAME)
T0 = datetime(2016, 11, 5, 23, 0, 0, 0)
T1 = datetime(2016, 11, 6, 6, 0, 0, 0)
T0_DST = datetime_with_timezone(datetime(2016, 11, 5, 23, 0, 0, 0),
time_zone=US_NY_ZONE_NAME)
T1_EST = datetime_with_timezone(datetime(2016, 11, 6, 6, 0, 0, 0),
time_zone=US_NY_ZONE_NAME)
def test_func_datetime_with_timezone1():
"""Test `datetime_with_timezone()`, DST.
"""
tt0 = datetime_with_timezone(T0_DST)
tt1 = datetime_with_timezone(T1_EST)
assert tt0.tzinfo.zone == US_NY_ZONE_NAME
assert bool(tt0.dst()) == True
assert bool(tt0.dst()) == is_dst(T0_DST)
assert tt1.tzinfo.zone == US_NY_ZONE_NAME