diff options
Diffstat (limited to '_test/lib/test_structure.py')
-rw-r--r-- | _test/lib/test_structure.py | 36 |
1 files changed, 21 insertions, 15 deletions
diff --git a/_test/lib/test_structure.py b/_test/lib/test_structure.py index 4d04b11..2656bbb 100644 --- a/_test/lib/test_structure.py +++ b/_test/lib/test_structure.py @@ -2,7 +2,7 @@ from __future__ import absolute_import from __future__ import print_function import ruamel.yaml as yaml -import canonical # NOQA +import canonical # NOQA import pprint from ruamel.yaml.compat import text_type, PY3 @@ -46,8 +46,12 @@ def test_structure(data_filename, structure_filename, verbose=False): with open(data_filename, 'rb') as fp: loader = yaml.Loader(fp) while loader.check_event(): - if loader.check_event(yaml.StreamStartEvent, yaml.StreamEndEvent, - yaml.DocumentStartEvent, yaml.DocumentEndEvent): + if loader.check_event( + yaml.StreamStartEvent, + yaml.StreamEndEvent, + yaml.DocumentStartEvent, + yaml.DocumentEndEvent, + ): loader.get_event() continue nodes1.append(_convert_structure(loader)) @@ -56,9 +60,9 @@ def test_structure(data_filename, structure_filename, verbose=False): assert nodes1 == nodes2, (nodes1, nodes2) finally: if verbose: - print("NODES1:") + print('NODES1:') pprint.pprint(nodes1) - print("NODES2:") + print('NODES2:') pprint.pprint(nodes2) @@ -89,9 +93,9 @@ def test_parser(data_filename, canonical_filename, verbose=False): _compare_events(events1, events2) finally: if verbose: - print("EVENTS1:") + print('EVENTS1:') pprint.pprint(events1) - print("EVENTS2:") + print('EVENTS2:') pprint.pprint(events2) @@ -109,9 +113,9 @@ def test_parser_on_canonical(canonical_filename, verbose=False): _compare_events(events1, events2, full=True) finally: if verbose: - print("EVENTS1:") + print('EVENTS1:') pprint.pprint(events1) - print("EVENTS2:") + print('EVENTS2:') pprint.pprint(events2) @@ -146,9 +150,9 @@ def test_composer(data_filename, canonical_filename, verbose=False): _compare_nodes(node1, node2) finally: if verbose: - print("NODES1:") + print('NODES1:') pprint.pprint(nodes1) - print("NODES2:") + print('NODES2:') pprint.pprint(nodes2) @@ -189,8 +193,9 @@ def _make_canonical_loader(): def construct_undefined(self, node): return self.construct_scalar(node) - MyCanonicalLoader.add_constructor(u'tag:yaml.org,2002:map', - MyCanonicalLoader.construct_mapping) + MyCanonicalLoader.add_constructor( + u'tag:yaml.org,2002:map', MyCanonicalLoader.construct_mapping + ) MyCanonicalLoader.add_constructor(None, MyCanonicalLoader.construct_undefined) @@ -207,9 +212,9 @@ def test_constructor(data_filename, canonical_filename, verbose=False): assert native1 == native2, (native1, native2) finally: if verbose: - print("NATIVE1:") + print('NATIVE1:') pprint.pprint(native1) - print("NATIVE2:") + print('NATIVE2:') pprint.pprint(native2) @@ -217,4 +222,5 @@ test_constructor.unittest = ['.data', '.canonical'] if __name__ == '__main__': import test_appliance + test_appliance.run(globals()) |