summaryrefslogtreecommitdiff
path: root/_test/lib/test_structure.py
diff options
context:
space:
mode:
Diffstat (limited to '_test/lib/test_structure.py')
-rw-r--r--_test/lib/test_structure.py36
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())