From 86437abb9f85397f79c5a98bc57212785b343aac Mon Sep 17 00:00:00 2001 From: Evandro Myller Date: Thu, 18 Sep 2025 16:05:15 -0300 Subject: [PATCH] Fix test data priority assignment --- tests/engine_tests/test_engine.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/engine_tests/test_engine.py b/tests/engine_tests/test_engine.py index 4d8023e..426abf4 100644 --- a/tests/engine_tests/test_engine.py +++ b/tests/engine_tests/test_engine.py @@ -91,10 +91,9 @@ def _extract_feature_contexts( key=itemgetter("id"), ) ] - if ( - priority := (feature_state.get("feature_segment") or {}).get("priority") - is not None - ): + if (feature_segment := feature_state.get("feature_segment")) and ( + priority := feature_segment.get("priority") + ) is not None: feature_context["priority"] = priority yield feature_context