test: correct yaml load warning

Author: woile

tests/commands/test_init_command.py

@@ -116,7 +116,9 @@ def test_no_existing_pre_commit_conifg(_, default_choice, tmpdir, config):
                 if "json" in default_choice:
                     assert json.load(file) == EXPECTED_DICT_CONFIG
                 elif "yaml" in default_choice:
-                    assert yaml.load(file) == EXPECTED_DICT_CONFIG
+                    assert (
+                        yaml.load(file, Loader=yaml.FullLoader) == EXPECTED_DICT_CONFIG
+                    )
                 else:
                     config_data = file.read()
                     assert config_data == expected_config
@@ -136,7 +138,9 @@ def test_empty_pre_commit_config(_, default_choice, tmpdir, config):
                 if "json" in default_choice:
                     assert json.load(file) == EXPECTED_DICT_CONFIG
                 elif "yaml" in default_choice:
-                    assert yaml.load(file) == EXPECTED_DICT_CONFIG
+                    assert (
+                        yaml.load(file, Loader=yaml.FullLoader) == EXPECTED_DICT_CONFIG
+                    )
                 else:
                     config_data = file.read()
                     assert config_data == expected_config
@@ -162,7 +166,9 @@ def test_pre_commit_config_without_cz_hook(_, default_choice, tmpdir, config):
                 if "json" in default_choice:
                     assert json.load(file) == EXPECTED_DICT_CONFIG
                 elif "yaml" in default_choice:
-                    assert yaml.load(file) == EXPECTED_DICT_CONFIG
+                    assert (
+                        yaml.load(file, Loader=yaml.FullLoader) == EXPECTED_DICT_CONFIG
+                    )
                 else:
                     config_data = file.read()
                     assert config_data == expected_config
@@ -184,7 +190,9 @@ def test_cz_hook_exists_in_pre_commit_config(_, default_choice, tmpdir, config):
                 if "json" in default_choice:
                     assert json.load(file) == EXPECTED_DICT_CONFIG
                 elif "yaml" in default_choice:
-                    assert yaml.load(file) == EXPECTED_DICT_CONFIG
+                    assert (
+                        yaml.load(file, Loader=yaml.FullLoader) == EXPECTED_DICT_CONFIG
+                    )
                 else:
                     config_data = file.read()
                     assert config_data == expected_config