diff options
Diffstat (limited to '_test/lib/test_emitter.py')
| -rw-r--r-- | _test/lib/test_emitter.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/_test/lib/test_emitter.py b/_test/lib/test_emitter.py index 39f0c1f..1158854 100644 --- a/_test/lib/test_emitter.py +++ b/_test/lib/test_emitter.py @@ -3,6 +3,7 @@ from __future__ import print_function import ruamel.yaml as yaml + def _compare_events(events1, events2): assert len(events1) == len(events2), (events1, events2) for event1, event2 in zip(events1, events2): @@ -16,6 +17,7 @@ def _compare_events(events1, events2): assert event1.tag == event2.tag, (event1, event2) assert event1.value == event2.value, (event1, event2) + def test_emitter_on_data(data_filename, canonical_filename, verbose=False): with open(data_filename, 'rb') as fp0: events = list(yaml.parse(fp0)) @@ -28,6 +30,7 @@ def test_emitter_on_data(data_filename, canonical_filename, verbose=False): test_emitter_on_data.unittest = ['.data', '.canonical'] + def test_emitter_on_canonical(canonical_filename, verbose=False): with open(canonical_filename, 'rb') as fp0: events = list(yaml.parse(fp0)) @@ -41,6 +44,7 @@ def test_emitter_on_canonical(canonical_filename, verbose=False): test_emitter_on_canonical.unittest = ['.canonical'] + def test_emitter_styles(data_filename, canonical_filename, verbose=False): for filename in [data_filename, canonical_filename]: with open(filename, 'rb') as fp0: @@ -51,13 +55,13 @@ def test_emitter_styles(data_filename, canonical_filename, verbose=False): for event in events: if isinstance(event, yaml.ScalarEvent): event = yaml.ScalarEvent(event.anchor, event.tag, - event.implicit, event.value, style=style) + event.implicit, event.value, style=style) elif isinstance(event, yaml.SequenceStartEvent): event = yaml.SequenceStartEvent(event.anchor, event.tag, - event.implicit, flow_style=flow_style) + event.implicit, flow_style=flow_style) elif isinstance(event, yaml.MappingStartEvent): event = yaml.MappingStartEvent(event.anchor, event.tag, - event.implicit, flow_style=flow_style) + event.implicit, flow_style=flow_style) styled_events.append(event) output = yaml.emit(styled_events) if verbose: @@ -69,6 +73,7 @@ def test_emitter_styles(data_filename, canonical_filename, verbose=False): test_emitter_styles.unittest = ['.data', '.canonical'] + class EventsLoader(yaml.Loader): def construct_event(self, node): @@ -77,7 +82,8 @@ class EventsLoader(yaml.Loader): else: mapping = self.construct_mapping(node) class_name = str(node.tag[1:])+'Event' - if class_name in ['AliasEvent', 'ScalarEvent', 'SequenceStartEvent', 'MappingStartEvent']: + if class_name in ['AliasEvent', 'ScalarEvent', 'SequenceStartEvent', + 'MappingStartEvent']: mapping.setdefault('anchor', None) if class_name in ['ScalarEvent', 'SequenceStartEvent', 'MappingStartEvent']: mapping.setdefault('tag', None) @@ -91,6 +97,7 @@ class EventsLoader(yaml.Loader): EventsLoader.add_constructor(None, EventsLoader.construct_event) + def test_emitter_events(events_filename, verbose=False): with open(events_filename, 'rb') as fp0: events = list(yaml.load(fp0, Loader=EventsLoader)) @@ -104,4 +111,3 @@ def test_emitter_events(events_filename, verbose=False): if __name__ == '__main__': import test_appliance test_appliance.run(globals()) - |