Changeset 8192 for main/waeup.kofa/trunk/src

Timestamp:

17 Apr 2012, 11:19:55 (13 years ago)

Author:

uli

Message:

Let to_timezone ignore not-datetimes.

Location:

main/waeup.kofa/trunk/src/waeup/kofa/utils

Files:

• helpers.py (modified) (1 diff)
• tests/test_helpers.py (modified) (1 diff)

main/waeup.kofa/trunk/src/waeup/kofa/utils/helpers.py

@@ -569 +569 @@
 
     If no `tz` is given, shift to UTC is performed.
-    """
+
+    If `dt` is not a datetime.datetime, the input value is returned
+    unchanged.
+    """
+    if not isinstance(dt, datetime.datetime):
+        return dt
     if tz is None:
         tz = pytz.utc

main/waeup.kofa/trunk/src/waeup/kofa/utils/tests/test_helpers.py

@@ -304 +304 @@
         return
 
+    def test_to_timezone_no_dt(self):
+        # the to_timezone function copes with dates (!= datetimes)
+        d = datetime.date(2012, 12, 1)
+        result1 = helpers.to_timezone(d)
+        result2 = helpers.to_timezone(d, pytz.utc)
+        self.assertEqual(result1, d)
+        self.assertEqual(result2, d)
+        return
 
 def test_suite():