From 56b78d4d1fd34d41eab473cf3d01946cc5b3c897 Mon Sep 17 00:00:00 2001 From: pratiksha badheka Date: Fri, 9 Jan 2026 01:38:57 -0800 Subject: [PATCH] update system path --- .../openlineage/tests/system/openlineage/conftest.py | 3 +-- .../openlineage/example_openlineage_base_complex_dag.py | 7 +++++-- .../openlineage/example_openlineage_base_simple_dag.py | 7 ++++--- .../openlineage/example_openlineage_defer_simple_dag.py | 7 ++++--- .../openlineage/example_openlineage_docs_file_dag.py | 7 ++++--- .../openlineage/example_openlineage_mapped_simple_dag.py | 7 ++++--- .../example_openlineage_schedule_asset_or_time_dag.py | 8 +++++--- .../openlineage/example_openlineage_schedule_cron_dag.py | 7 ++++--- ...xample_openlineage_schedule_list_complex_assets_dag.py | 8 +++++--- ...ample_openlineage_schedule_list_multiple_assets_dag.py | 7 ++++--- .../example_openlineage_schedule_list_single_asset_dag.py | 7 ++++--- .../example_openlineage_schedule_multiple_assets_dag.py | 7 ++++--- .../example_openlineage_schedule_single_asset_dag.py | 7 ++++--- .../example_openlineage_schedule_timetable_dag.py | 7 ++++--- .../openlineage/example_openlineage_setup_teardown_dag.py | 7 ++++--- .../openlineage/example_openlineage_short_circuit_dag.py | 7 ++++--- .../openlineage/example_openlineage_task_groups_dag.py | 6 ++++-- .../example_openlineage_taskflow_simple_dag.py | 6 ++++-- .../system/openlineage/example_openlineage_trigger_dag.py | 7 ++++--- .../example_openlineage_trigger_dag_deferrable.py | 7 ++++--- .../openlineage/example_openlineage_versioned_dag.py | 7 ++++--- 21 files changed, 84 insertions(+), 59 deletions(-) diff --git a/providers/openlineage/tests/system/openlineage/conftest.py b/providers/openlineage/tests/system/openlineage/conftest.py index aa3b87ad5530a..640ddd938d12d 100644 --- a/providers/openlineage/tests/system/openlineage/conftest.py +++ b/providers/openlineage/tests/system/openlineage/conftest.py @@ -19,8 +19,7 @@ import pytest from airflow.providers.openlineage.plugins.listener import OpenLineageListener - -from system.openlineage.transport.variable import VariableTransport +from airflow.providers.openlineage.tests.system.openlineage.transport.variable import VariableTransport @pytest.fixture(autouse=True) diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_base_complex_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_base_complex_dag.py index de1636e1b9108..7c03b07f7c199 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_base_complex_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_base_complex_dag.py @@ -46,8 +46,11 @@ except ImportError: from airflow.utils.task_group import TaskGroup # type: ignore[no-redef] -from system.openlineage.expected_events import AIRFLOW_VERSION, get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + AIRFLOW_VERSION, + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator def check_events_number_func(): diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_base_simple_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_base_simple_dag.py index 74986bdd0f4f6..6320170ead703 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_base_simple_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_base_simple_dag.py @@ -28,11 +28,12 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - try: from airflow.sdk import chain except ImportError: diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_defer_simple_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_defer_simple_dag.py index 145fdca650555..e9c876f3345cf 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_defer_simple_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_defer_simple_dag.py @@ -28,12 +28,13 @@ from airflow import DAG from airflow.models import Variable +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator from airflow.providers.standard.sensors.time_delta import TimeDeltaSensor -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - def check_events_number_func(): events = Variable.get(key="openlineage_defer_simple_dag.wait.event.start", deserialize_json=True) diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_docs_file_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_docs_file_dag.py index 067a4937563e2..6939e1547f78f 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_docs_file_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_docs_file_dag.py @@ -26,11 +26,12 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_docs_file_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_simple_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_simple_dag.py index a80608edd233b..f9cad0bae627b 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_simple_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_simple_dag.py @@ -35,11 +35,12 @@ # Airflow 2 path from airflow.decorators import task # type: ignore[attr-defined,no-redef] from airflow.models import Variable +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - def check_events_number_func(): for event_type in ("start", "complete"): diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_asset_or_time_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_asset_or_time_dag.py index 7c173b3790195..1446bc2e91e93 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_asset_or_time_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_asset_or_time_dag.py @@ -27,12 +27,14 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + AIRFLOW_VERSION, + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator from airflow.timetables.trigger import CronTriggerTimetable -from system.openlineage.expected_events import AIRFLOW_VERSION, get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_asset_or_time_dag" if AIRFLOW_VERSION.major == 3: diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_cron_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_cron_dag.py index 77750f760c397..dec9fa05fc3a6 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_cron_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_cron_dag.py @@ -26,11 +26,12 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_cron_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_complex_assets_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_complex_assets_dag.py index c3e38ee235879..253a446ead1d9 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_complex_assets_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_complex_assets_dag.py @@ -28,11 +28,13 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + AIRFLOW_VERSION, + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import AIRFLOW_VERSION, get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - if AIRFLOW_VERSION.major == 3: schedule = [ (Asset(uri="s3://bucket/file.txt", extra={"a": 1}) | Asset(uri="s3://bucket2/file.txt")) diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_multiple_assets_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_multiple_assets_dag.py index 8997ac02dfa7a..e85593bba7626 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_multiple_assets_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_multiple_assets_dag.py @@ -27,11 +27,12 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_list_multiple_assets_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_single_asset_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_single_asset_dag.py index db0aa95527e48..de81f914a01e0 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_single_asset_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_list_single_asset_dag.py @@ -27,11 +27,12 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_list_single_asset_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_multiple_assets_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_multiple_assets_dag.py index a8228ca09f1da..c2c1ce0dab85e 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_multiple_assets_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_multiple_assets_dag.py @@ -27,11 +27,12 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_multiple_assets_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_single_asset_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_single_asset_dag.py index 40b0a1d6cbf9e..e1da3c93223b9 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_single_asset_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_single_asset_dag.py @@ -27,11 +27,12 @@ from airflow import DAG from airflow.providers.common.compat.assets import Asset +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_single_asset_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_timetable_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_timetable_dag.py index 7e552374e2d21..d831853e914ce 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_timetable_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_schedule_timetable_dag.py @@ -28,12 +28,13 @@ import pendulum from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator from airflow.timetables.events import EventsTimetable -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_schedule_timetable_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_setup_teardown_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_setup_teardown_dag.py index 6e56c34dc13b4..4da39db7216c6 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_setup_teardown_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_setup_teardown_dag.py @@ -27,11 +27,12 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - def do_nothing(): pass diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_short_circuit_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_short_circuit_dag.py index 4f0048d479c93..f6d94f5123876 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_short_circuit_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_short_circuit_dag.py @@ -28,11 +28,12 @@ from airflow import DAG from airflow.models import Variable +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator, ShortCircuitOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - def do_nothing(): pass diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_task_groups_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_task_groups_dag.py index cde217b62ae72..b4f303fa9e769 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_task_groups_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_task_groups_dag.py @@ -33,8 +33,10 @@ except ImportError: from airflow.utils.task_group import TaskGroup # type: ignore[no-redef] -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator DAG_ID = "openlineage_task_groups_dag" diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_taskflow_simple_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_taskflow_simple_dag.py index 4b7fda60f5dfd..38193bf851ef5 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_taskflow_simple_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_taskflow_simple_dag.py @@ -27,8 +27,10 @@ from datetime import datetime -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator try: from airflow.sdk import dag, task diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag.py index fb524344bab4d..8ccc51d3e3d1e 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag.py @@ -28,12 +28,13 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator from airflow.providers.standard.operators.trigger_dagrun import TriggerDagRunOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_trigger_dag" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag_deferrable.py b/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag_deferrable.py index 4a4f8ee934bd8..1cfd4c001d28e 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag_deferrable.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_trigger_dag_deferrable.py @@ -29,12 +29,13 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.bash import BashOperator from airflow.providers.standard.operators.trigger_dagrun import TriggerDagRunOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - DAG_ID = "openlineage_trigger_dag_deferrable" with DAG( diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_versioned_dag.py b/providers/openlineage/tests/system/openlineage/example_openlineage_versioned_dag.py index a36c65d7e717d..165f9fdaa91c7 100644 --- a/providers/openlineage/tests/system/openlineage/example_openlineage_versioned_dag.py +++ b/providers/openlineage/tests/system/openlineage/example_openlineage_versioned_dag.py @@ -28,11 +28,12 @@ from datetime import datetime from airflow import DAG +from airflow.providers.openlineage.tests.system.openlineage.expected_events import ( + get_expected_event_file_path, +) +from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator from airflow.providers.standard.operators.python import PythonOperator -from system.openlineage.expected_events import get_expected_event_file_path -from system.openlineage.operator import OpenLineageTestOperator - def do_nothing(): pass