Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
pankajastro committed Jul 23, 2024
1 parent cc16402 commit 94ae3f7
Showing 1 changed file with 8 additions and 11 deletions.
19 changes: 8 additions & 11 deletions tests/test_example_dags.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
except ImportError:
from functools import lru_cache as cache


import airflow
import pytest
from airflow.models.dagbag import DagBag
Expand All @@ -31,7 +30,6 @@

IGNORED_DAG_FILES = ["performance_dag.py"]


# Sort descending based on Versions and convert string to an actual version
MIN_VER_DAG_FILE_VER: dict[Version, list[str]] = {
Version(version): MIN_VER_DAG_FILE[version] for version in sorted(MIN_VER_DAG_FILE, key=Version, reverse=True)
Expand Down Expand Up @@ -88,15 +86,14 @@ def get_dag_ids() -> list[str]:
AIRFLOW_VERSION in PARTIALLY_SUPPORTED_AIRFLOW_VERSIONS,
reason="Airflow 2.9.0 and 2.9.1 have a breaking change in Dataset URIs, and Cosmos errors if `emit_datasets` is not False",
)
@pytest.mark.integration
@pytest.mark.parametrize("dag_id", get_dag_ids())
def test_example_dag(session, dag_id: str):
if dag_id == "jaffle_shop_kubernetes":
return
dag_bag = get_dag_bag()
dag = dag_bag.get_dag(dag_id)
test_utils.run_dag(dag)

# @pytest.mark.integration
# @pytest.mark.parametrize("dag_id", get_dag_ids())
# def test_example_dag(session, dag_id: str):
# if dag_id == "jaffle_shop_kubernetes":
# return
# dag_bag = get_dag_bag()
# dag = dag_bag.get_dag(dag_id)
# test_utils.run_dag(dag)

@pytest.mark.integration
def test_example_dag_k8s(session):
Expand Down

0 comments on commit 94ae3f7

Please sign in to comment.