diff --git a/tests/unittest/test_convert_to_markdown.py b/tests/unittest/test_convert_to_markdown.py
index f3c02d870..5d9142030 100644
--- a/tests/unittest/test_convert_to_markdown.py
+++ b/tests/unittest/test_convert_to_markdown.py
@@ -53,7 +53,7 @@ def test_simple_dictionary_input(self):
'relevant_line': '[return ""](https://github.com/Codium-ai/pr-agent-pro/pull/102/files#diff-52d45f12b836f77ed1aef86e972e65404634ea4e2a6083fb71a9b0f9bb9e062fR199)'}]}
- expected_output = f'{PRReviewHeader.REGULAR} 🔍\n\n
\n⏱️ Estimated effort to review: 1 🔵⚪⚪⚪⚪ |
\n🧪 No relevant tests |
\n⚡ Possible issues: No\n |
\n🔒 No security concerns identified |
\n
\n\n\n Code feedback:
\n\n
relevant file | pr_agent/git_providers/git_provider.py\n |
suggestion | \n\n\n\nConsider raising an exception or logging a warning when \'pr_url\' attribute is not found. This can help in debugging issues related to the absence of \'pr_url\' in instances where it\'s expected. [important]\n\n\n |
relevant line | return "" |
\n\n '
+ expected_output = f'{PRReviewHeader.REGULAR.value} 🔍\n\n\n⏱️ Estimated effort to review: 1 🔵⚪⚪⚪⚪ |
\n🧪 No relevant tests |
\n⚡ Possible issues: No\n |
\n🔒 No security concerns identified |
\n
\n\n\n Code feedback:
\n\n
relevant file | pr_agent/git_providers/git_provider.py\n |
suggestion | \n\n\n\nConsider raising an exception or logging a warning when \'pr_url\' attribute is not found. This can help in debugging issues related to the absence of \'pr_url\' in instances where it\'s expected. [important]\n\n\n |
relevant line | return "" |
\n\n '
assert convert_to_markdown_v2(input_data).strip() == expected_output.strip()
diff --git a/tests/unittest/test_load_yaml.py b/tests/unittest/test_load_yaml.py
index 9314c89f9..90c4c196e 100644
--- a/tests/unittest/test_load_yaml.py
+++ b/tests/unittest/test_load_yaml.py
@@ -45,7 +45,7 @@ def test_load_invalid_yaml2(self):
with pytest.raises(ScannerError):
yaml.safe_load(yaml_str)
- expected_output = [{'relevant file': 'src/app.py:', 'suggestion content': 'The print statement is outside inside the if __name__ ==: '}]
+ expected_output = [{'relevant file': 'src/app.py:', 'suggestion content': 'The print statement is outside inside the if __name__ ==:'}]
assert load_yaml(yaml_str) == expected_output