From 09ede2753e1ce67e72ad23d118c7a96048cbd2e0 Mon Sep 17 00:00:00 2001 From: Jens Scheffler Date: Sat, 15 Feb 2025 18:45:08 +0100 Subject: [PATCH] Remove assert in example DAG --- .../airflow/providers/edge/example_dags/integration_test.py | 3 ++- pyproject.toml | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/providers/edge/src/airflow/providers/edge/example_dags/integration_test.py b/providers/edge/src/airflow/providers/edge/example_dags/integration_test.py index 69d8dabdc0f96..a850e5dae0853 100644 --- a/providers/edge/src/airflow/providers/edge/example_dags/integration_test.py +++ b/providers/edge/src/airflow/providers/edge/example_dags/integration_test.py @@ -99,7 +99,8 @@ def virtualenv(): @task def variable(): Variable.set("integration_test_key", "value") - assert Variable.get("integration_test_key") == "value" + if Variable.get("integration_test_key") != "value": + raise ValueError("Variable not set as expected.") Variable.delete("integration_test_key") @task diff --git a/pyproject.toml b/pyproject.toml index 52b5827ae8365..6ea5e71dffb14 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -374,8 +374,6 @@ testing = ["dev", "providers.tests", "task_sdk.tests", "tests_common", "tests"] "kubernetes_tests/*" = ["D", "TID253", "S101", "TRY002"] "helm_tests/*" = ["D", "TID253", "S101", "TRY002"] "providers/**/tests/*" = ["D", "TID253", "S101", "TRY002"] -# exclude assert in edge provider -"providers/edge/src/airflow/providers/edge/example_dags/integration_test.py" = ["S101"] # All of the modules which have an extra license header (i.e. that we copy from another project) need to # ignore E402 -- module level import not at top level