kaizenkodo/ChatMastermind
2
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases Wiki Activity

Fix tests.

Browse Source
This commit is contained in:
Oleksandr Kozachuk 2023-08-12 13:17:10 +02:00
parent 1fb9144192
commit 6ed459be6f
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_main.py
View File

@ -102,6 +102,9 @@ class TestHandleQuestion(unittest.TestCase):
source=None, source=None,
only_source_code=False, only_source_code=False,
number=3, number=3,
max_tokens=None,
temperature=None,
model=None,
match_all_tags=False, match_all_tags=False,
with_tags=False, with_tags=False,
with_file=False, with_file=False,
@ -109,18 +112,19 @@ class TestHandleQuestion(unittest.TestCase):
self.config = { self.config = {
'db': 'test_files', 'db': 'test_files',
'setting1': 'value1', 'setting1': 'value1',
'setting2': 'value2' 'setting2': 'value2',
'openai': {},
} }
@patch("chatmastermind.main.create_chat_hist", return_value="test_chat") @patch("chatmastermind.main.create_chat_hist", return_value="test_chat")
@patch("chatmastermind.main.print_tag_args") @patch("chatmastermind.main.print_tag_args")
@patch("chatmastermind.utils.print_chat_hist") @patch("chatmastermind.main.print_chat_hist")
@patch("chatmastermind.main.ai", return_value=(["answer1", "answer2", "answer3"], "test_usage")) @patch("chatmastermind.main.ai", return_value=(["answer1", "answer2", "answer3"], "test_usage"))
@patch("chatmastermind.utils.pp") @patch("chatmastermind.utils.pp")
@patch("builtins.print") @patch("builtins.print")
def test_ask_cmd(self, mock_print, mock_pp, mock_ai, def test_ask_cmd(self, mock_print, mock_pp, mock_ai,
mock_print_tag_args, mock_create_chat_hist, mock_print_chat_hist, mock_print_tag_args,
mock_print_chat_hist): mock_create_chat_hist):
open_mock = MagicMock() open_mock = MagicMock()
with patch("chatmastermind.storage.open", open_mock): with patch("chatmastermind.storage.open", open_mock):
ask_cmd(self.args, self.config) ask_cmd(self.args, self.config)
@ -135,7 +139,6 @@ class TestHandleQuestion(unittest.TestCase):
mock_print_chat_hist.assert_called_once_with('test_chat', mock_print_chat_hist.assert_called_once_with('test_chat',
False, False,
self.args.only_source_code) self.args.only_source_code)
mock_pp.assert_called_once_with("test_chat")
mock_ai.assert_called_with("test_chat", mock_ai.assert_called_with("test_chat",
self.config, self.config,
self.args.number) self.args.number)