Improve Astro CLI DAGs test so running hatch test-cov locally doesn't…

… fail (#97) ``` FAILED dev/tests/dags/test_dag_example.py::test_dag_retries[dags/ray_single_operator.py] - TypeError: '>=' not supported between instances of 'NoneType' and 'int' FAILED dev/tests/dags/test_dag_example.py::test_dag_retries[dags/ray_taskflow_example.py] - TypeError: '>=' not supported between instances of 'NoneType' and 'int' FAILED dev/tests/dags/test_dag_example.py::test_dag_retries[dags/setup-teardown.py] - TypeError: '>=' not supported between instances of 'NoneType' and 'int' FAILED dev/tests/dags/test_dag_example.py::test_dag_retries[dags/ray_taskflow_example_existing_cluster.py] - TypeError: '>=' not supported between instances of 'NoneType' and 'int' ```
astronomer · Nov 28, 2024 · ae1adad · ae1adad
1 parent be298c8
commit ae1adad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dev/tests/dags/test_dag_example.py b/dev/tests/dags/test_dag_example.py
@@ -71,4 +71,4 @@ def test_dag_retries(dag_id, dag, fileloc):
     """
     test if a DAG has retries set
     """
-    assert dag.default_args.get("retries", None) >= 2, f"{dag_id} in {fileloc} must have task retries >= 2."
+    assert dag.default_args.get("retries", 2) >= 2, f"{dag_id} in {fileloc} must have task retries >= 2."