Fixed DST handling for from_date/to_date, closes #193 (again)

tests/test_datetime_utils.py

@@ -0,0 +1,90 @@
+""" test datetime_utils """
+import pytest
+
+
+def test_get_local_tz():
+    """ test get_local_tz during time with no DST """
+    import datetime
+    import os
+    import time
+
+    from osxphotos.datetime_utils import get_local_tz
+
+    os.environ["TZ"] = "US/Pacific"
+    time.tzset()
+
+    dt = datetime.datetime(2018, 12, 31, 0, 0, 0)
+    local_tz = get_local_tz(dt)
+    assert local_tz == datetime.timezone(
+        datetime.timedelta(days=-1, seconds=57600), "PST"
+    )
+
+
+def test_get_local_tz_dst():
+    """ test get_local_tz during time with DST """
+    import datetime
+    import os
+    import time
+
+    from osxphotos.datetime_utils import get_local_tz
+
+    os.environ["TZ"] = "US/Pacific"
+    time.tzset()
+
+    dt = datetime.datetime(2018, 6, 30, 0, 0, 0)
+    local_tz = get_local_tz(dt)
+    assert local_tz == datetime.timezone(
+        datetime.timedelta(days=-1, seconds=61200), "PDT"
+    )
+
+
+def test_datetime_remove_tz():
+    """ test datetime_remove_tz """
+    import datetime
+
+    from osxphotos.datetime_utils import datetime_remove_tz
+
+    dt = datetime.datetime(
+        2018,
+        12,
+        31,
+        tzinfo=datetime.timezone(datetime.timedelta(days=-1, seconds=57600), "PST"),
+    )
+    dt_no_tz = datetime_remove_tz(dt)
+    assert dt_no_tz.tzinfo is None
+
+
+def test_datetime_has_tz():
+    """ test datetime_has_tz """
+    import datetime
+
+    from osxphotos.datetime_utils import datetime_has_tz
+
+    dt = datetime.datetime(
+        2018,
+        12,
+        31,
+        tzinfo=datetime.timezone(datetime.timedelta(days=-1, seconds=57600), "PST"),
+    )
+    assert datetime_has_tz(dt)
+
+    dt_notz = datetime.datetime(2018, 12, 31)
+    assert not datetime_has_tz(dt_notz)
+
+
+def test_datetime_naive_to_local():
+    """ test datetime_naive_to_local """
+    import datetime
+    import os
+    import time
+
+    from osxphotos.datetime_utils import datetime_naive_to_local
+
+    os.environ["TZ"] = "US/Pacific"
+    time.tzset()
+
+    dt = datetime.datetime(2018, 6, 30, 0, 0, 0)
+    dt_local = datetime_naive_to_local(dt)
+    assert dt_local.tzinfo == datetime.timezone(
+        datetime.timedelta(days=-1, seconds=61200), "PDT"
+    )