kyfx
/
pytest-bdd
mirror of https://github.com/pytest-dev/pytest-bdd.git
5
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Make mypy happy

This commit is contained in:
Jason Allen 2024-11-05 19:04:34 +00:00
parent 186c666857
commit 1e53a6ee62
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/pytest_bdd/parser.py
View File

@ -4,7 +4,7 @@ import os.path
import re import re
import textwrap import textwrap
from collections import OrderedDict from collections import OrderedDict
from collections.abc import Iterable, Mapping, Sequence from collections.abc import Generator, Iterable, Mapping, Sequence
from dataclasses import dataclass, field from dataclasses import dataclass, field
from typing import Any from typing import Any
@ -13,6 +13,7 @@ from .gherkin_parser import Background as GherkinBackground
from .gherkin_parser import DataTable from .gherkin_parser import DataTable
from .gherkin_parser import Feature as GherkinFeature from .gherkin_parser import Feature as GherkinFeature
from .gherkin_parser import GherkinDocument from .gherkin_parser import GherkinDocument
from .gherkin_parser import Rule as GherkinRule
from .gherkin_parser import Scenario as GherkinScenario from .gherkin_parser import Scenario as GherkinScenario
from .gherkin_parser import Step as GherkinStep from .gherkin_parser import Step as GherkinStep
from .gherkin_parser import Tag as GherkinTag from .gherkin_parser import Tag as GherkinTag
@ -499,7 +500,7 @@ class FeatureParser:
return feature return feature
def _parse_and_add_rule(self, rule_data, feature): def _parse_and_add_rule(self, rule_data: GherkinRule, feature: Feature) -> None:
"""Parse a rule, including its background and scenarios, and add to the feature.""" """Parse a rule, including its background and scenarios, and add to the feature."""
rule = Rule( rule = Rule(
keyword=rule_data.keyword, keyword=rule_data.keyword,
@ -519,7 +520,7 @@ class FeatureParser:
scenario = self.parse_scenario(rule_scenario, feature, rule) scenario = self.parse_scenario(rule_scenario, feature, rule)
feature.scenarios[scenario.name] = scenario feature.scenarios[scenario.name] = scenario
def _extract_rule_background(self, rule_data, rule): def _extract_rule_background(self, rule_data: GherkinRule, rule: Rule) -> Background | None:
"""Extract the first background from rule children if it exists.""" """Extract the first background from rule children if it exists."""
for child in rule_data.children: for child in rule_data.children:
if child.background: if child.background:
@ -527,13 +528,15 @@ class FeatureParser:
return None return None
@staticmethod @staticmethod
def _extract_rule_scenarios(rule_data): def _extract_rule_scenarios(rule_data: GherkinRule) -> Generator[GherkinScenario]:
"""Yield each scenario under a rule.""" """Yield each scenario under a rule."""
for child in rule_data.children: for child in rule_data.children:
if child.scenario: if child.scenario:
yield child.scenario yield child.scenario
def _parse_and_add_scenario(self, scenario_data, feature, rule=None): def _parse_and_add_scenario(
self, scenario_data: GherkinScenario, feature: Feature, rule: Rule | None = None
) -> None:
"""Parse an individual scenario and add it to the feature's scenarios.""" """Parse an individual scenario and add it to the feature's scenarios."""
scenario = self.parse_scenario(scenario_data, feature, rule) scenario = self.parse_scenario(scenario_data, feature, rule)
feature.scenarios[scenario.name] = scenario feature.scenarios[scenario.name] = scenario