diff --git a/oyaml.py b/oyaml.py index dfc7977..0fdf8e4 100644 --- a/oyaml.py +++ b/oyaml.py @@ -18,6 +18,7 @@ def map_constructor(loader, node): pyyaml.add_representer(dict, map_representer) pyyaml.add_representer(OrderedDict, map_representer) +pyyaml.add_representer(dict, map_representer, Dumper=pyyaml.dumper.SafeDumper) pyyaml.add_representer(OrderedDict, map_representer, Dumper=pyyaml.dumper.SafeDumper) diff --git a/setup.py b/setup.py index 5aa46b7..0d8f6fb 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name='oyaml', - version='0.3', + version='0.4', description='Ordered YAML: drop-in replacement for PyYAML which preserves dict ordering', long_description=open('README.rst').read(), author='Wim Glenn', diff --git a/test_oyaml.py b/test_oyaml.py index 80368c6..69ff6d1 100644 --- a/test_oyaml.py +++ b/test_oyaml.py @@ -3,7 +3,6 @@ from types import GeneratorType import pytest -from yaml.representer import RepresenterError import oyaml as yaml @@ -23,6 +22,12 @@ def test_dump_all(): assert yaml.dump_all(documents=[data, {}]) == '{x: 1, z: 3, y: 2}\n--- {}\n' +def test_dump_and_safe_dump_match(): + mydict = {'x': 1, 'z': 2, 'y': 3} + # don't know if mydict is ordered in the implementation or not (but don't care) + assert yaml.dump(mydict) == yaml.safe_dump(mydict) + + def test_safe_dump_all(): assert yaml.safe_dump_all(documents=[data, {}]) == '{x: 1, z: 3, y: 2}\n--- {}\n' @@ -60,7 +65,7 @@ class MyOrderedDict(OrderedDict): data = MyOrderedDict([('x', 1), ('y', 2)]) assert '!!python/object/apply:test_oyaml.MyOrderedDict' in yaml.dump(data) - with pytest.raises(RepresenterError) as cm: + with pytest.raises(yaml.pyyaml.representer.RepresenterError) as cm: yaml.safe_dump(data) assert str(cm.value) == "cannot represent an object: MyOrderedDict([('x', 1), ('y', 2)])"