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

Small fixes.

Browse Source
This commit is contained in:
Oleksandr Kozachuk
2023-09-09 16:05:27 +02:00
parent d22877a0f1
commit 39b518a8a6
4 changed files with 19 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_question_cmd.py
+7 -7
View File
@@ -24,7 +24,7 @@ class TestMessageCreate(unittest.TestCase):
self.args = MagicMock(spec=argparse.Namespace)
self.args.source = None
self.args.source_code = None
self.args.ai = None
self.args.AI = None
self.args.model = None
self.args.output_tags = None
# create some files for sourcing
@@ -59,7 +59,7 @@ Language is called 'brainfart'."""
return list(Path(tmp_dir.name).glob('*.[ty]*'))
def test_message_file_created(self) -> None:
self.args.question = ["What is this?"]
self.args.ask = ["What is this?"]
cache_dir_files = self.message_list(self.cache_path)
self.assertEqual(len(cache_dir_files), 0)
create_message(self.chat, self.args)
@@ -70,14 +70,14 @@ Language is called 'brainfart'."""
self.assertEqual(message.question, Question("What is this?")) # type: ignore [union-attr]
def test_single_question(self) -> None:
self.args.question = ["What is this?"]
self.args.ask = ["What is this?"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)
self.assertEqual(message.question, Question("What is this?"))
self.assertEqual(len(message.question.source_code()), 0)
def test_multipart_question(self) -> None:
self.args.question = ["What is this", "'bard' thing?", "Is it good?"]
self.args.ask = ["What is this", "'bard' thing?", "Is it good?"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)
self.assertEqual(message.question, Question("""What is this
@@ -87,7 +87,7 @@ Language is called 'brainfart'."""
Is it good?"""))
def test_single_question_with_text_only_source(self) -> None:
self.args.question = ["What is this?"]
self.args.ask = ["What is this?"]
self.args.source = [f"{self.source_file1.name}"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)
@@ -99,7 +99,7 @@ Is it good?"""))
{self.source_file1_content}"""))
def test_single_question_with_embedded_source_source(self) -> None:
self.args.question = ["What is this?"]
self.args.ask = ["What is this?"]
self.args.source = [f"{self.source_file2.name}"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)
@@ -111,7 +111,7 @@ Is it good?"""))
{self.source_file2_content}"""))
def test_single_question_with_embedded_source_code_source(self) -> None:
self.args.question = ["What is this?"]
self.args.ask = ["What is this?"]
self.args.source_code = [f"{self.source_file2.name}"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)