From 5af9e8e74998746e83d305dd8ebc118116a5cbfc Mon Sep 17 00:00:00 2001 From: "Hussam.lawen" Date: Thu, 8 Feb 2024 23:53:29 +0200 Subject: [PATCH] fix --- pr_agent/algo/utils.py | 2 +- tests/unittest/test_load_yaml.py | 17 ++++++++--------- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/pr_agent/algo/utils.py b/pr_agent/algo/utils.py index 7b189fca..d2a08a04 100644 --- a/pr_agent/algo/utils.py +++ b/pr_agent/algo/utils.py @@ -333,7 +333,7 @@ def load_yaml(response_text: str, keys_fix_yaml: List[str] = []) -> dict: def try_fix_yaml(response_text: str, keys_fix_yaml: List[str] = []) -> dict: response_text_lines = response_text.split('\n') - keys = ['relevant line:', 'suggestion content:', 'relevant_file:', 'existing code:', 'improved code:'] + keys = ['relevant line:', 'suggestion content:', 'relevant file:', 'existing code:', 'improved code:'] keys = keys + keys_fix_yaml # first fallback - try to convert 'relevant line: ...' to relevant line: |-\n ...' response_text_lines_copy = response_text_lines.copy() diff --git a/tests/unittest/test_load_yaml.py b/tests/unittest/test_load_yaml.py index 55bc1205..5736ae3b 100644 --- a/tests/unittest/test_load_yaml.py +++ b/tests/unittest/test_load_yaml.py @@ -38,16 +38,15 @@ PR Feedback: assert load_yaml(yaml_str) == expected_output def test_load_invalid_yaml2(self): - yaml_str = \ -'''\ - - relevant file: src/app.py: - suggestion content: The print statement is outside inside the if __name__ ==: ''' - with pytest.raises(ScannerError): - yaml.safe_load(yaml_str) + yaml_str = '''\ +- relevant file: src/app.py: + suggestion content: The print statement is outside inside the if __name__ ==: \ +''' + 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__ ==: '}] - assert load_yaml(yaml_str) == expected_output + 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