add tzinfo

pynamodb · Oct 14, 2020 · 1b5e085 · 1b5e085
1 parent 9ff4ebf
commit 1b5e085
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_attributes.py b/tests/test_attributes.py
@@ -884,7 +884,7 @@ class CreatedAtMap(DynamicMapAttribute):
 
     def test_serialize(self):
         test_model = TestDynamicMapAttribute.TestModel()
-        test_model.test_map.created_at = datetime(2017, 1, 1)
+        test_model.test_map.created_at = datetime(2017, 1, 1, tzinfo=timezone.utc)
         test_model.test_map.foo = 'bar'
         test_model.test_map.empty = None
         assert test_model.serialize() == {'test_map': {'M': {
@@ -901,7 +901,7 @@ def test_deserialize(self):
         }}}
         test_model = TestDynamicMapAttribute.TestModel()
         test_model.deserialize(serialized)
-        assert test_model.test_map.created_at == datetime(2017, 1, 1)
+        assert test_model.test_map.created_at == datetime(2017, 1, 1, tzinfo=timezone.utc)
         assert test_model.test_map.foo == 'bar'
         assert test_model.test_map.empty is None