Ensures UTC stays a singleton

Author: rozza

bson/tz_util.py

@@ -38,6 +38,9 @@ def __init__(self, offset, name):
     def __getinitargs__(self):
         return self.__offset, self.__name
 
+    def __reduce__(self):
+        return _UTC, ()
+
     def utcoffset(self, dt):
         return self.__offset
 
@@ -50,3 +53,12 @@ def dst(self, dt):
 
 utc = FixedOffset(0, "UTC")
 """Fixed offset timezone representing UTC."""
+
+
+def _UTC():
+    """Factory function for utc unpickling.
+
+    Makes sure that unpickling a utc instance always returns the same
+    module global.
+    """
+    return utc

test/test_timestamp.py

@@ -48,9 +48,11 @@ def test_datetime_copy_pickle(self):
 
         dc = copy.deepcopy(d)
         self.assertEqual(dc, t.as_datetime())
+        self.assertEqual(d.tzinfo, dc.tzinfo)
 
         dp = pickle.loads(pickle.dumps(d))
         self.assertEqual(dp, t.as_datetime())
+        self.assertEqual(d.tzinfo, dp.tzinfo)
 
     def test_exceptions(self):
         self.assertRaises(TypeError, Timestamp)