More toml timezone fixes

bkabrda · Mar 21, 2017 · efc4b3f · efc4b3f
1 parent 9f04528
commit efc4b3f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_parse.py b/test/test_parse.py
@@ -5,6 +5,7 @@
 
 import pytest
 import six
+import toml
 
 from anymarkup_core import *
 
@@ -89,7 +90,8 @@ def test_parse_force_types_false(self, str, fmt, expected):
         (types_ini, None, {'x': {'a': '1', 'b': '1.1', 'c': 'None', 'd': 'True'}}),
         (types_json, None, {'x': {'a': 1, 'b': 1.1, 'c': None, 'd': True}}),
         (types_json5, 'json5', {'x': {'a': 1, 'b': 1.1, 'c': None, 'd': True}}),
-        (types_toml, 'toml', {'x': {'a': 1, 'b': 1.1, 'c': datetime(1987, 7, 5, 17, 45),
+        (types_toml, 'toml', {'x': {'a': 1, 'b': 1.1,
+                                    'c': datetime(1987, 7, 5, 17, 45, tzinfo=TomlTz('Z')),
                                     'd': True}}),
         (types_xml, None, {'x': {'a': '1', 'b': '1.1', 'c': 'None', 'd': 'True'}}),
         (types_yaml, None, {'x': {'a': 1, 'b': 1.1, 'c': 'None', 'd': True}}),