Add new YAML processing tests in test_load_yaml.py for snippet extrac…

…tion and line removal

tests/test_load_yaml.py

@@ -92,3 +92,16 @@ def test_load_invalid_yaml2(self):
             )
             == expected_output
         )
+
+# auto-generated by cover agent
+def test_try_fix_yaml_snippet_extraction():
+    from cover_agent.utils import try_fix_yaml
+    yaml_str = "```yaml\nname: John Smith\nage: 35\n```"
+    expected_output = {"name": "John Smith", "age": 35}
+    assert try_fix_yaml(yaml_str) == expected_output
+
+def test_try_fix_yaml_remove_all_lines():
+    from cover_agent.utils import try_fix_yaml
+    yaml_str = "name: John Smith\nage: 35\ninvalid_line"
+    expected_output = {"name": "John Smith", "age": 35}
+    assert try_fix_yaml(yaml_str) == expected_output