Allow in question -s for just sourcing file and -S to source file with ``` encapsulation.

This commit is contained in:
Oleksandr Kozachuk
2023-09-09 15:16:17 +02:00
parent 3eca53998b
commit 86eebc39ea
3 changed files with 32 additions and 17 deletions
+18 -4
View File
@@ -23,7 +23,7 @@ class TestMessageCreate(unittest.TestCase):
# create arguments mock
self.args = MagicMock(spec=argparse.Namespace)
self.args.source = None
self.args.source_code_only = False
self.args.source_code = None
self.args.ai = None
self.args.model = None
self.args.output_tags = None
@@ -94,11 +94,11 @@ Is it good?"""))
# source file contains no source code
# -> don't expect any in the question
self.assertEqual(len(message.question.source_code()), 0)
self.assertEqual(message.question, Question("""What is this?
self.assertEqual(message.question, Question(f"""What is this?
{self.source_file1_content}"""))
def test_single_question_with_embedded_source_code_source(self) -> None:
def test_single_question_with_embedded_source_source(self) -> None:
self.args.question = ["What is this?"]
self.args.source = [f"{self.source_file2.name}"]
message = create_message(self.chat, self.args)
@@ -106,6 +106,20 @@ Is it good?"""))
# source file 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?
self.assertEqual(message.question, Question(f"""What is this?
{self.source_file2_content}"""))
def test_single_question_with_embedded_source_code_source(self) -> None:
self.args.question = ["What is this?"]
self.args.source_code = [f"{self.source_file2.name}"]
message = create_message(self.chat, self.args)
self.assertIsInstance(message, Message)
# source file contains 1 source code block
# -> expect it in the question
self.assertEqual(len(message.question.source_code()), 2)
self.assertEqual(message.question, Question(f"""What is this?
```
{self.source_file2_content}
```"""))