diff options
Diffstat (limited to '_test/lib/test_canonical.py')
-rw-r--r-- | _test/lib/test_canonical.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/_test/lib/test_canonical.py b/_test/lib/test_canonical.py index 797dddc..33387cd 100644 --- a/_test/lib/test_canonical.py +++ b/_test/lib/test_canonical.py @@ -1,13 +1,14 @@ -#from __future__ import absolute_import +# from __future__ import absolute_import from __future__ import print_function import ruamel.yaml -import canonical +import canonical # NOQA + def test_canonical_scanner(canonical_filename, verbose=False): with open(canonical_filename, 'rb') as fp0: data = fp0.read() - tokens = list(yaml.canonical_scan(data)) + tokens = list(ruamel.yaml.canonical_scan(data)) assert tokens, tokens if verbose: for token in tokens: @@ -15,10 +16,11 @@ def test_canonical_scanner(canonical_filename, verbose=False): test_canonical_scanner.unittest = ['.canonical'] + def test_canonical_parser(canonical_filename, verbose=False): with open(canonical_filename, 'rb') as fp0: data = fp0.read() - events = list(yaml.canonical_parse(data)) + events = list(ruamel.yaml.canonical_parse(data)) assert events, events if verbose: for event in events: @@ -26,12 +28,13 @@ def test_canonical_parser(canonical_filename, verbose=False): test_canonical_parser.unittest = ['.canonical'] + def test_canonical_error(data_filename, canonical_filename, verbose=False): with open(data_filename, 'rb') as fp0: data = fp0.read() try: - output = list(yaml.canonical_load_all(data)) - except yaml.YAMLError as exc: + output = list(ruamel.yaml.canonical_load_all(data)) # NOQA + except ruamel.yaml.YAMLError as exc: if verbose: print(exc) else: @@ -43,4 +46,3 @@ test_canonical_error.skip = ['.empty'] if __name__ == '__main__': import test_appliance test_appliance.run(globals()) - |