test: multi-line f-string removal overextension; fix: same; fix: remo…

…val of colon

tests/more_languages/group7/dataclass.py

@@ -22,3 +22,15 @@ def execute(self, *args, **kwargs): ...
     output_model: DataClass
 
     def execute(self, *args, **kwargs): ...
+
+    @property
+    def edge_case(self) -> str:
+        if not self.name and not self.description:
+            return ""
+        return f"""Location: {self.config.name}
+
+{self.config.description}
+"""
+
+    def should_still_see_me(self, x: bool = True) -> "Tool":
+        pass

tests/test_more_language_units.py

@@ -2072,6 +2072,9 @@ def test_more_languages_tensorflow_flags():
     """@dataclass(frozen=True, slots=True, kw_only=True)
 class Tool(Protocol)""",
     "    def execute(self, *args, **kwargs)",
+    """    @property
+    def edge_case(self) -> str""",
+    '    def should_still_see_me(self, x: bool = True) -> "Tool"',
 ]
 
 WGSL_EXPECTATION = [

tree_plus_src/parse_file.py

@@ -1042,15 +1042,11 @@ def remove_py_comments(input_string: str) -> str:
 # from pygments.lexers.python import PythonLexer
 
 
-docstring_pattern = re.compile(r"^\s*\"\"\"[\s\S]+?\"\"\"", re.MULTILINE)
+docstring_pattern = re.compile(r"(?::)\s^\s+\"\"\"[\s\S]+?\"\"\"", re.MULTILINE)
 
 
 def remove_docstrings(source):
-    """
-    Remove docstrings from a source code string.
-    """
-    # Matches triple-quoted strings (single or double quotes)
-    return docstring_pattern.sub("", source)
+    return docstring_pattern.sub(":", source)
 
 
 def parse_py(contents: str) -> List[str]: