@@ -90,12 +90,8 @@ def test_add_str_to_type_other_type(self) -> None:
9090 def test_enhance_docstring_for_references (self ) -> None :
9191 """Test docstring enhancement for referenceable functions."""
9292 original = "Original docstring."
93- param_info = {
94- "data" : {"original" : "dict" , "modified" : "dict | str" , "accepts_str" : False },
95- "text" : {"original" : "str" , "modified" : "str" , "accepts_str" : True },
96- }
9793
98- result = _enhance_docstring_for_references (original , param_info , "test_func" )
94+ result = _enhance_docstring_for_references (original , "test_func" )
9995
10096 assert "Original docstring." in result
10197 assert "All parameters accept object references" in result
@@ -106,7 +102,7 @@ def test_enhance_docstring_for_references(self) -> None:
106102
107103 def test_enhance_docstring_for_references_empty_original (self ) -> None :
108104 """Test docstring enhancement with empty original docstring."""
109- result = _enhance_docstring_for_references ("" , {}, "test_func" )
105+ result = _enhance_docstring_for_references ("" , "test_func" )
110106
111107 assert "test_func function." in result
112108 assert "All parameters accept object references" in result
@@ -148,7 +144,7 @@ def test_enhance_docstring_for_references_preserves_original(self) -> None:
148144 :raises RuntimeError: If processing fails
149145 """
150146
151- result = _enhance_docstring_for_references (original , {}, "test_func" )
147+ result = _enhance_docstring_for_references (original , "test_func" )
152148
153149 # Check that all original content is preserved exactly
154150 assert "Process data and return results." in result
0 commit comments