From 97f0a2315736e50f1b34a015447cd751da66ecb6 Mon Sep 17 00:00:00 2001 From: Dirk Mueller Date: Mon, 25 Jan 2021 22:49:04 +0100 Subject: [PATCH] Use Yaml's UnsafeLoader for Python embedding tests In newer PyYAML versions the default FullLoader has python/object/* integration removed. One has to use UnsafeLoader instead. see this issue for details: https://github.com/yaml/pyyaml/issues/321 --- test/test_example.py | 2 +- test/test_functional.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/test/test_example.py b/test/test_example.py index db2cf17..1d27043 100644 --- a/test/test_example.py +++ b/test/test_example.py @@ -151,7 +151,7 @@ if have_yaml_support: @ddt class YamlOnlyTestCase(unittest.TestCase): - @file_data('data/test_custom_yaml_loader.yaml', yaml.FullLoader) + @file_data('data/test_custom_yaml_loader.yaml', yaml.UnsafeLoader) def test_custom_yaml_loader(self, instance, expected): """Test with yaml tags to create specific classes to compare""" self.assertEqual(expected, instance) diff --git a/test/test_functional.py b/test/test_functional.py index 45c1cae..f9609c6 100644 --- a/test/test_functional.py +++ b/test/test_functional.py @@ -440,7 +440,7 @@ def test_load_yaml_with_python_tag(): loader allowing python tags is passed. """ - from yaml import FullLoader + from yaml import UnsafeLoader from yaml.constructor import ConstructorError def str_to_type(class_name): @@ -457,13 +457,13 @@ def test_load_yaml_with_python_tag(): raise AssertionError() @ddt - class YamlFullLoaderTest(object): - @file_data('data/test_functional_custom_tags.yaml', FullLoader) + class YamlUnsafeLoaderTest(object): + @file_data('data/test_functional_custom_tags.yaml', UnsafeLoader) def test_cls_is_instance(self, instance, expected): assert isinstance(instance, str_to_type(expected)) - tests = list(filter(_is_test, YamlFullLoaderTest.__dict__)) - obj = YamlFullLoaderTest() + tests = list(filter(_is_test, YamlUnsafeLoaderTest.__dict__)) + obj = YamlUnsafeLoaderTest() if not tests: raise AssertionError('No tests have been found.') -- 2.29.2