diff --git a/tests/test_importers.py b/tests/test_importers.py index bb8757c9c33b65b7a1d8a4d6acd77205465e8fd1..01d66e6a9a46264869b9aa0c8fba4b1d64d6d579 100644 --- a/tests/test_importers.py +++ b/tests/test_importers.py @@ -14,9 +14,14 @@ from import2ledger import importers, strparse from . import DATA_DIR +try: + load_yaml = yaml.full_load +except AttributeError: + load_yaml = yaml.load + class TestImporters: with pathlib.Path(DATA_DIR, 'imports.yml').open() as yaml_file: - test_data = yaml.load(yaml_file) + test_data = load_yaml(yaml_file) for test in test_data: test['source'] = DATA_DIR / test['source']