Allow in question -s for just sourcing file and -S to source file with ``` encapsulation.
This commit is contained in:
@@ -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}
|
||||
```"""))
|
||||
|
||||
Reference in New Issue
Block a user