question_cmd: now also accepts Messages as source files

This commit is contained in:
juk0de 2023-09-13 14:59:29 +02:00
parent 8e5da48f4d
commit be8747c8ad

View File

@ -181,14 +181,15 @@ This is embedded source code.
def test_single_question_with_message_and_embedded_code(self) -> None:
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)
self.assertIsInstance(message, Message)
# file contains no source code (only text)
# -> don't expect any in the question
# answer contains 1 source code block
# -> expect it in the question
self.assertEqual(len(message.question.source_code()), 1)
self.assertEqual(message.question, Question("""What is this?
```
It is embedded code
```"""))
```
"""))