From 764cc49c5347213fd26f1ee1bbfdc9f2f9729ee5 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Sun, 18 Dec 2016 02:01:17 -0500 Subject: [PATCH] test_machine_parse --- maya.py | 5 +++++ test_maya.py | 7 +++++++ 2 files changed, 12 insertions(+) diff --git a/maya.py b/maya.py index b132ac6..94d52e5 100644 --- a/maya.py +++ b/maya.py @@ -62,6 +62,11 @@ class MayaDT(object): @staticmethod def __dt_to_epoch(dt): """Converts a datetime into an epoch.""" + + # Assume UTC if no datetime is provided. + if dt.tzinfo is None: + dt = dt.replace(tzinfo=pytz.utc) + epoch_start = Datetime(*EPOCH_START, tzinfo=pytz.timezone('UTC')) return (dt - epoch_start).total_seconds() diff --git a/test_maya.py b/test_maya.py index 1fc35a7..8406b6c 100644 --- a/test_maya.py +++ b/test_maya.py @@ -19,6 +19,13 @@ def test_human_when(): assert r2.day - r1.day == 1 +def test_machine_parse(): + r1 = maya.parse('August 14, 2015') + assert r1.day == 14 + + r2 = maya.parse('August 15, 2015') + assert r2.day == 15 + def test_dt_tz_translation(): d1 = maya.now().datetime()