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

question_cmd: now also accepts Messages as source files

Browse Source
This commit is contained in:
juk0de 2023-09-13 14:59:29 +02:00
parent 8e5da48f4d
commit be8747c8ad
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_question_cmd.py
View File

@ -181,14 +181,15 @@ This is embedded source code.
def test_single_question_with_message_and_embedded_code(self) -> None: def test_single_question_with_message_and_embedded_code(self) -> None:
self.args.ask = ["What is this?"] self.args.ask = ["What is this?"]
self.args.source_text = [f"{self.chat.messages[1].file_path}"] self.args.source_code = [f"{self.chat.messages[1].file_path}"]
message = create_message(self.chat, self.args) message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message) self.assertIsInstance(message, Message)
# file contains no source code (only text) # answer contains 1 source code block
# -> don't expect any in the question # -> expect it in the question
self.assertEqual(len(message.question.source_code()), 1) self.assertEqual(len(message.question.source_code()), 1)
self.assertEqual(message.question, Question("""What is this? self.assertEqual(message.question, Question("""What is this?
``` ```
It is embedded code It is embedded code
```""")) ```
"""))