Merge pull request andersinno#25 from suutari-ai/yaml-safe-load

Use yaml.safe_load to read the configuration

Author: suutari-ai

database_sanitizer/config.py

@@ -48,7 +48,7 @@ def from_file(cls, filename):
         instance = cls()
 
         with open(filename, "rb") as file_stream:
-            config_data = yaml.load(file_stream)
+            config_data = yaml.safe_load(file_stream)
 
         instance.load(config_data)
 

database_sanitizer/tests/test_config.py

@@ -10,7 +10,7 @@
 
 
 @mock.patch.object(config, 'open')
-@mock.patch('yaml.load')
+@mock.patch('yaml.safe_load')
 def test_from_file(mocked_yaml_load, mocked_open):
     mocked_yaml_load.return_value = {}