diff --git a/features/steps/actor_compute_impact_error_handling_steps.py b/features/steps/actor_compute_impact_error_handling_steps.py index 306bacc2d..05ca2e4f6 100644 --- a/features/steps/actor_compute_impact_error_handling_steps.py +++ b/features/steps/actor_compute_impact_error_handling_steps.py @@ -192,9 +192,7 @@ def step_action_count(context: Any, count: int) -> None: def step_warning_log_contains(context: Any, text: str) -> None: """Assert that at least one WARNING log message contains the given text.""" warning_messages = [ - r.getMessage() - for r in context.log_records - if r.levelno >= logging.WARNING + r.getMessage() for r in context.log_records if r.levelno >= logging.WARNING ] assert any(text in msg for msg in warning_messages), ( f"Expected a WARNING log message containing {text!r}, " @@ -206,9 +204,7 @@ def step_warning_log_contains(context: Any, text: str) -> None: def step_no_warning_logs(context: Any) -> None: """Assert that no WARNING log messages were emitted.""" warning_messages = [ - r.getMessage() - for r in context.log_records - if r.levelno >= logging.WARNING + r.getMessage() for r in context.log_records if r.levelno >= logging.WARNING ] assert not warning_messages, ( f"Expected no WARNING log messages, but got: {warning_messages}"