| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253 |
- # -*- coding: utf-8 -*-
- import pytest
- from ioex import AutoDict
- yaml = pytest.importorskip('yaml')
- def test_to_yaml():
- d = AutoDict()
- d['A'] = 1.1
- d['B'][2] = 'b'
- d['C'][3]['c'] = [u'Γ', u'γ']
- class TestDumper(yaml.SafeDumper):
- pass
- TestDumper.add_representer(type(d), lambda d, c: type(c).to_yaml(d, c))
- generated_yaml = yaml.dump(
- d,
- Dumper=TestDumper,
- default_flow_style=False)
- loaded_dict = yaml.load(generated_yaml)
- assert isinstance(loaded_dict, dict)
- expected_dict = {
- 'A': 1.1,
- 'B': {2: 'b'},
- 'C': {3: {'c': [u'Γ', u'γ']}},
- }
- assert expected_dict == loaded_dict
- def test_register_yaml_representer():
- d = AutoDict()
- d['A'] = 1.1
- d['B'][2] = 'b'
- d['C'][3]['c'] = [u'Γ', u'γ']
- class TestDumper(yaml.SafeDumper):
- pass
- AutoDict.register_yaml_representer(TestDumper)
- generated_yaml = yaml.dump(
- d,
- Dumper=TestDumper,
- default_flow_style=False)
- loaded_dict = yaml.load(generated_yaml)
- assert isinstance(loaded_dict, dict)
- expected_dict = {
- 'A': 1.1,
- 'B': {2: 'b'},
- 'C': {3: {'c': [u'Γ', u'γ']}},
- }
- assert expected_dict == loaded_dict
|
