Use yaml.safe_load instead of load (encode#6719)

rhcarvalho · rpkilby · commit 62ed1f827079 · 2019-05-29T09:51:24.000-07:00
Use of PyYAML's yaml.load function without specifying the Loader parameter has been deprecated, see https://msg.pyyaml.org/load. Earlier versions of PyYAML already had the alternative safe_load function, which limits the loader to a subset of YAML constructs, that is enough for what we need here. Fixes encode#6677
diff --git a/tests/schemas/test_managementcommand.py b/tests/schemas/test_managementcommand.py
@@ -45,7 +45,7 @@ def test_renders_default_schema_with_custom_title_url_and_description(self):
                      '--description=Sample description',
                      stdout=self.out)
         # Check valid YAML was output.
-        schema = yaml.load(self.out.getvalue())
+        schema = yaml.safe_load(self.out.getvalue())
         assert schema['openapi'] == '3.0.2'
 
     def test_renders_openapi_json_schema(self):
