format

apache · Feb 6, 2025 · 92458a8 · 92458a8
1 parent 29e6c49
commit 92458a8
Showing 1 changed file with 18 additions and 9 deletions.
diff --git a/tests/charts/log_groomer.py b/tests/charts/log_groomer.py
@@ -181,9 +181,13 @@ def test_log_groomer_retention_days_overrides(self, retention_days, retention_re
         )
 
         if retention_result:
-            assert jmespath.search(
-                "spec.template.spec.containers[1].env[?name=='AIRFLOW__LOG_RETENTION_DAYS'].value | [0]", docs[0]
-            ) == retention_result
+            assert (
+                jmespath.search(
+                    "spec.template.spec.containers[1].env[?name=='AIRFLOW__LOG_RETENTION_DAYS'].value | [0]",
+                    docs[0],
+                )
+                == retention_result
+            )
         else:
             assert len(jmespath.search("spec.template.spec.containers[1].env", docs[0])) == 2
 
@@ -205,9 +209,13 @@ def test_log_groomer_frequency_minutes_overrides(self, frequency_minutes, freque
         )
 
         if frequency_result:
-            assert jmespath.search(
-                "spec.template.spec.containers[1].env[?name=='AIRFLOW__LOG_CLEANUP_FREQUENCY_MINUTES'].value | [0]", docs[0]
-            ) == frequency_result
+            assert (
+                jmespath.search(
+                    "spec.template.spec.containers[1].env[?name=='AIRFLOW__LOG_CLEANUP_FREQUENCY_MINUTES'].value | [0]",
+                    docs[0],
+                )
+                == frequency_result
+            )
         else:
             assert len(jmespath.search("spec.template.spec.containers[1].env", docs[0])) == 2
 
@@ -262,6 +270,7 @@ def test_log_groomer_has_airflow_home(self):
             values=values, show_only=[f"templates/{self.folder}/{self.obj_name}-deployment.yaml"]
         )
 
-        assert jmespath.search(
-            "spec.template.spec.containers[1].env[?name=='AIRFLOW_HOME'].name | [0]", docs[0]
-        ) == "AIRFLOW_HOME"
+        assert (
+            jmespath.search("spec.template.spec.containers[1].env[?name=='AIRFLOW_HOME'].name | [0]", docs[0])
+            == "AIRFLOW_HOME"
+        )