added testcases for messages.py

chatmastermind/message.py

@@ -22,13 +22,16 @@ def source_code(text: str, include_delims: bool = False) -> list[str]:
     code_lines: list[str] = []
     in_code_block = False
 
+    print(text)
+
     for line in text.split('\n'):
-        if line.strip().startswith('```'):
+        if line.strip().startswith("```"):
             if include_delims:
                 code_lines.append(line)
             if in_code_block:
-                code_sections.append('\n'.join(code_lines) + '\n')
+                code_sections.append("\n".join(code_lines))
                 code_lines.clear()
+                print(f" --> New code section:\n{code_sections}")
             in_code_block = not in_code_block
         elif in_code_block:
             code_lines.append(line)

tests/test_main.py

@@ -363,8 +363,8 @@ class SourceCodeTestCase(CmmTestCase):
         ```
         """
         expected_result = [
-            "        ```python\n        print(\"Hello, World!\")\n        ```\n",
-            "        ```python\n        x = 10\n        y = 20\n        print(x + y)\n        ```\n"
+            "       ```python\n     print(\"Hello, World!\")\n      ```",
+            "       ```python\n     x = 10\n        y = 20\n        print(x + y)\n```"
         ]
         result = source_code(text, include_delims=True)
         self.assertEqual(result, expected_result)
@@ -383,15 +383,15 @@ class SourceCodeTestCase(CmmTestCase):
         ```
         """
         expected_result = [
-            "        print(\"Hello, World!\")\n",
-            "        x = 10\n        y = 20\n        print(x + y)\n"
+            "       print(\"Hello, World!\")\n",
+            "       x = 10\n        y = 20\n        print(x + y\n)"
         ]
         result = source_code(text, include_delims=False)
         self.assertEqual(result, expected_result)
 
     def test_source_code_with_single_code_block(self) -> None:
         text = "```python\nprint(\"Hello, World!\")\n```"
-        expected_result = ["```python\nprint(\"Hello, World!\")\n```\n"]
+        expected_result = ["```python\nprint(\"Hello, World!\")\n```"]
         result = source_code(text, include_delims=True)
         self.assertEqual(result, expected_result)