Add tests

tanloong · tanloong · commit e4ef49de7a0a · 2025-05-10T04:55:53.000+08:00
diff --git a/Lib/test/test_pyrepl/test_pyrepl.py b/Lib/test/test_pyrepl/test_pyrepl.py
@@ -605,6 +605,43 @@ def test_auto_indent_ignore_comments(self):
         output = multiline_input(reader)
         self.assertEqual(output, output_code)
 
+    def test_auto_indent_noncomment_hash(self):
+        # fmt: off
+        events = code_to_events(
+            "if ' ' == '#':\n"
+                "pass\n\n"
+        )
+
+        output_code = (
+            "if ' ' == '#':\n"
+            "    pass\n"
+            "    "
+        )
+        # fmt: on
+
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, output_code)
+
+    def test_auto_indent_multiline_string(self):
+        # fmt: off
+        events = code_to_events(
+            "s = '''\n"
+            "Note:\n"
+            "'''\n\n"
+        )
+
+        output_code = (
+            "s = '''\n"
+            "Note:\n"
+            "'''"
+        )
+        # fmt: on
+
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, output_code)
+
 
 class TestPyReplOutput(ScreenEqualMixin, TestCase):
     def prepare_reader(self, events):
