Add test

tanloong · tanloong · commit dee83020f76a · 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
@@ -572,6 +572,46 @@ def test_auto_indent_with_comment(self):
         output = multiline_input(reader)
         self.assertEqual(output, output_code)
 
+        # fmt: off
+        events = code_to_events(
+            "def f():\n"
+                "# foo\n"
+                "pass\n\n"
+        )
+
+        output_code = (
+            "def f():\n"
+            "    # foo\n"
+            "    pass\n"
+            "    "
+        )
+        # fmt: on
+
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, output_code)
+
+        # fmt: off
+        events = itertools.chain(
+            code_to_events("def f():\n"),
+            [
+                Event(evt="key", data="backspace", raw=b"\x08"),
+            ],
+            code_to_events("# foo\npass\n\n")
+        )
+
+        output_code = (
+            "def f():\n"
+            "# foo\n"
+            "    pass\n"
+            "    "
+        )
+        # fmt: on
+
+        reader = self.prepare_reader(events)
+        output = multiline_input(reader)
+        self.assertEqual(output, output_code)
+
     def test_auto_indent_with_multicomment(self):
         # fmt: off
         events = code_to_events(
@@ -605,7 +645,7 @@ def test_auto_indent_ignore_comments(self):
         output = multiline_input(reader)
         self.assertEqual(output, output_code)
 
-    def test_auto_indent_noncomment_hash(self):
+    def test_auto_indent_hashtag(self):
         # fmt: off
         events = code_to_events(
             "if ' ' == '#':\n"
