Fix accessing item.obj (mypy was rightfully complaining)

pytest-dev · Jan 21, 2024 · 4dcc2c7 · 4dcc2c7
1 parent e28774c
commit 4dcc2c7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/pytest_bdd/generation.py b/src/pytest_bdd/generation.py
@@ -6,6 +6,7 @@
 from typing import TYPE_CHECKING, cast
 
 from _pytest._io import TerminalWriter
+from _pytest.python import Function
 from mako.lookup import TemplateLookup
 
 from .feature import get_features
@@ -183,6 +184,8 @@ def _show_missing_code_main(config: Config, session: Session) -> None:
     features, scenarios, steps = parse_feature_files(config.option.features)
 
     for item in session.items:
+        if not isinstance(item, Function):
+            continue
         if (scenario := scenario_wrapper_template_registry.get(item.obj)) is not None:
             if scenario in scenarios:
                 scenarios.remove(scenario)