Fix list_py_file_paths in test

apache · Jan 18, 2025 · db90345 · db90345
1 parent e50fe58
commit db90345
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/api_fastapi/core_api/routes/public/test_dag_report.py b/tests/api_fastapi/core_api/routes/public/test_dag_report.py
@@ -35,7 +35,9 @@
 
 
 def get_corresponding_dag_file_count(dir: str, include_examples: bool = True) -> int:
-    return len(list_py_file_paths(directory=dir, include_examples=include_examples))
+    return len(list_py_file_paths(directory=dir)) + (
+        len(list_py_file_paths("/opt/airflow/airflow/example_dags")) if include_examples else 0
+    )
 
 
 class TestDagReportEndpoint: