diff options
Diffstat (limited to 'tests/lib/test_errors.py')
| -rw-r--r-- | tests/lib/test_errors.py | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tests/lib/test_errors.py b/tests/lib/test_errors.py index 9ef9bbe..45f233f 100644 --- a/tests/lib/test_errors.py +++ b/tests/lib/test_errors.py @@ -3,7 +3,8 @@ import yaml, test_emitter def test_loader_error(error_filename, verbose=False): try: - list(yaml.load_all(open(error_filename, 'rb'), yaml.FullLoader)) + with open(error_filename, 'rb') as file: + list(yaml.load_all(file, yaml.FullLoader)) except yaml.YAMLError as exc: if verbose: print("%s:" % exc.__class__.__name__, exc) @@ -14,7 +15,8 @@ test_loader_error.unittest = ['.loader-error'] def test_loader_error_string(error_filename, verbose=False): try: - list(yaml.load_all(open(error_filename, 'rb').read(), yaml.FullLoader)) + with open(error_filename, 'rb') as file: + list(yaml.load_all(file.read(), yaml.FullLoader)) except yaml.YAMLError as exc: if verbose: print("%s:" % exc.__class__.__name__, exc) @@ -25,7 +27,8 @@ test_loader_error_string.unittest = ['.loader-error'] def test_loader_error_single(error_filename, verbose=False): try: - yaml.load(open(error_filename, 'rb').read(), yaml.FullLoader) + with open(error_filename, 'rb') as file: + yaml.load(file.read(), yaml.FullLoader) except yaml.YAMLError as exc: if verbose: print("%s:" % exc.__class__.__name__, exc) @@ -35,8 +38,8 @@ def test_loader_error_single(error_filename, verbose=False): test_loader_error_single.unittest = ['.single-loader-error'] def test_emitter_error(error_filename, verbose=False): - events = list(yaml.load(open(error_filename, 'rb'), - Loader=test_emitter.EventsLoader)) + with open(error_filename, 'rb') as file: + events = list(yaml.load(file, Loader=test_emitter.EventsLoader)) try: yaml.emit(events) except yaml.YAMLError as exc: @@ -48,7 +51,8 @@ def test_emitter_error(error_filename, verbose=False): test_emitter_error.unittest = ['.emitter-error'] def test_dumper_error(error_filename, verbose=False): - code = open(error_filename, 'rb').read() + with open(error_filename, 'rb') as file: + code = file.read() try: import yaml from io import StringIO |