Move cloned entries declaration out of raise assertion scope

Signed-off-by: Hai Nguyen <[email protected]>
apache · Mar 18, 2024 · 21f4e9a · 21f4e9a
1 parent 0d4ae03
commit 21f4e9a
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py b/tests/providers/cncf/kubernetes/operators/test_custom_object_launcher.py
@@ -77,28 +77,29 @@ def test_spark_job_spec_dynamicAllocation_enabled_with_invalid_config(self):
                 "executor": {},
             }
         }
+
+        cloned_entries = entries.copy()
+        cloned_entries["spec"]["dynamicAllocation"]["initialExecutors"] = None
         with pytest.raises(
             AirflowException,
             match="Make sure initial/min/max value for dynamic allocation is passed",
         ):
-            cloned_entries = entries.copy()
-            cloned_entries["spec"]["dynamicAllocation"]["initialExecutors"] = None
             SparkJobSpec(**cloned_entries)
 
+        cloned_entries = entries.copy()
+        cloned_entries["spec"]["dynamicAllocation"]["minExecutors"] = None
         with pytest.raises(
             AirflowException,
             match="Make sure initial/min/max value for dynamic allocation is passed",
         ):
-            cloned_entries = entries.copy()
-            cloned_entries["spec"]["dynamicAllocation"]["minExecutors"] = None
             SparkJobSpec(**cloned_entries)
 
+        cloned_entries = entries.copy()
+        cloned_entries["spec"]["dynamicAllocation"]["maxExecutors"] = None
         with pytest.raises(
             AirflowException,
             match="Make sure initial/min/max value for dynamic allocation is passed",
         ):
-            cloned_entries = entries.copy()
-            cloned_entries["spec"]["dynamicAllocation"]["maxExecutors"] = None
             SparkJobSpec(**cloned_entries)