Changeset 225 for pyyaml/trunk/tests/test_constructor.py

Timestamp:

08/16/06 14:22:38 (7 years ago)

Author:

xi

Message:

Fix timestamp constructing and representing (close #25).

File:

• pyyaml/trunk/tests/test_constructor.py (modified) (4 diffs)

pyyaml/trunk/tests/test_constructor.py

@@ -1 +1 @@
 
 import test_appliance
-try:
-    import datetime
-except ImportError:
-    pass
+
+import datetime
 try:
     set
@@ -240 +238 @@
     def __eq__(self, other):
         return type(self) is type(other) and dict(self) == dict(other)
+
+class FixedOffset(datetime.tzinfo):
+
+    def __init__(self, offset, name):
+        self.__offset = datetime.timedelta(minutes=offset)
+        self.__name = name
+
+    def utcoffset(self, dt):
+        return self.__offset
+
+    def tzname(self, dt):
+        return self.__name
+
+    def dst(self, dt):
+        return datetime.timedelta(0)
+
 
 def execute(code):
@@ -258 +272 @@
             try:
                 self.failUnlessEqual(data1, data2)
-            except AssertionError:
+            except (AssertionError, TypeError):
                 if isinstance(data1, dict):
                     data1 = [(repr(key), value) for key, value in data1.items()]
@@ -275 +289 @@
                                 (item2 != item2 or (item2 == 0.0 and item2 == 1.0)):
                             continue
+                        if isinstance(item1, datetime.datetime):
+                            item1 = item1.utctimetuple()
+                        if isinstance(item2, datetime.datetime):
+                            item2 = item2.utctimetuple()
                         self.failUnlessEqual(item1, item2)
                 else: