Skip to content

Commit 4472511

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent ca43222 commit 4472511

File tree

1 file changed

+15
-34
lines changed

1 file changed

+15
-34
lines changed
Lines changed: 15 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
from unittest.mock import patch, mock_open
1+
from unittest.mock import mock_open, patch
2+
23
import pytest
34

45
from ai_feedback.__main__ import _load_content_with_fallback
@@ -11,66 +12,46 @@ def test_load_content_predefined_name_success(self):
1112
"""Test loading content using a predefined name."""
1213
content = "Test prompt content"
1314
predefined_values = ["test_prompt", "another_prompt"]
14-
15+
1516
with patch("builtins.open", mock_open(read_data=content)):
1617
with patch("os.path.join") as mock_join:
1718
mock_join.return_value = "/fake/path/user/test_prompt.md"
18-
19-
result = _load_content_with_fallback(
20-
"test_prompt",
21-
predefined_values,
22-
"user",
23-
"prompt"
24-
)
25-
19+
20+
result = _load_content_with_fallback("test_prompt", predefined_values, "user", "prompt")
21+
2622
assert result == content
2723
mock_join.assert_called_once()
2824

2925
def test_load_content_custom_file_path_success(self):
3026
"""Test loading content using a custom file path."""
3127
content = "Custom file content"
3228
predefined_values = ["predefined_prompt"]
33-
29+
3430
with patch("builtins.open", mock_open(read_data=content)):
35-
result = _load_content_with_fallback(
36-
"/custom/path/file.md",
37-
predefined_values,
38-
"user",
39-
"prompt"
40-
)
41-
31+
result = _load_content_with_fallback("/custom/path/file.md", predefined_values, "user", "prompt")
32+
4233
assert result == content
4334

4435
def test_load_content_predefined_name_not_found(self, capsys):
4536
"""Test error when predefined file is not found."""
4637
predefined_values = ["test_prompt"]
47-
38+
4839
with patch("builtins.open", side_effect=FileNotFoundError):
4940
with pytest.raises(SystemExit) as exc_info:
50-
_load_content_with_fallback(
51-
"test_prompt",
52-
predefined_values,
53-
"user",
54-
"prompt"
55-
)
56-
41+
_load_content_with_fallback("test_prompt", predefined_values, "user", "prompt")
42+
5743
assert exc_info.value.code == 1
5844
captured = capsys.readouterr()
5945
assert "Pre-defined prompt file 'test_prompt.md' not found in user subfolder." in captured.out
6046

6147
def test_load_content_custom_file_path_not_found(self, capsys):
6248
"""Test error when custom file is not found."""
6349
predefined_values = ["predefined_prompt"]
64-
50+
6551
with patch("builtins.open", side_effect=FileNotFoundError):
6652
with pytest.raises(SystemExit) as exc_info:
67-
_load_content_with_fallback(
68-
"/nonexistent/file.md",
69-
predefined_values,
70-
"user",
71-
"prompt"
72-
)
73-
53+
_load_content_with_fallback("/nonexistent/file.md", predefined_values, "user", "prompt")
54+
7455
assert exc_info.value.code == 1
7556
captured = capsys.readouterr()
7657
assert "Prompt file '/nonexistent/file.md' not found." in captured.out

0 commit comments

Comments
 (0)