use yaml.BaseLoader in test to remove warning

sayint-ai · Dec 12, 2019 · 154d367 · 154d367
1 parent a649697
commit 154d367
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/api/test_responses.py b/tests/api/test_responses.py
@@ -61,7 +61,7 @@ def test_openapi_yaml_behind_proxy(reverse_proxied_app):
     )
     assert openapi_yaml.status_code == 200
     assert openapi_yaml.headers.get('Content-Type') == 'text/yaml'
-    spec = yaml.load(openapi_yaml.data.decode('utf-8'))
+    spec = yaml.load(openapi_yaml.data.decode('utf-8'), Loader=yaml.BaseLoader)
 
     if reverse_proxied_app._spec_file == 'swagger.yaml':
         assert b'url = "/behind/proxy/v1.0/swagger.json"' in swagger_ui.data