more robust tests

theislab · Jan 12, 2024 · b0cf09b · b0cf09b
1 parent 6f3dcd4
commit b0cf09b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_elegant_typehints.py b/tests/test_elegant_typehints.py
@@ -95,7 +95,10 @@ def process(fn: Callable[..., Any], *, run_napoleon: bool = False) -> list[str]:
         else:
             name = "???"
         for listener in listeners:
-            if not run_napoleon and listener.handler.__name__ == "_process_docstring":
+            if (
+                not run_napoleon
+                and listener.handler.__module__ == "sphinx.ext.napoleon"
+            ):
                 continue
             listener.handler(app, "function", name, fn, None, lines)
         return lines