diff options
Diffstat (limited to '_test')
-rw-r--r-- | _test/test_z_data.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/_test/test_z_data.py b/_test/test_z_data.py index ca75ca6..99810aa 100644 --- a/_test/test_z_data.py +++ b/_test/test_z_data.py @@ -9,6 +9,7 @@ import warnings # NOQA from ruamel.std.pathlib import Path base_path = Path('data') # that is ruamel.yaml.data +PY2 = sys.version_info[0] == 2 class YAMLData(object): @@ -124,7 +125,10 @@ class TestYAMLData(object): buf = StringIO() yaml.dump(data, buf) expected = input.value if output is None else output.value - assert buf.getvalue() == expected + value = buf.getvalue() + if PY2: + value = value.decode('utf-8') + assert value == expected def load_assert(self, input, confirm, yaml_version=None): from ruamel.yaml.compat import Mapping |