From 56b78d4d1fd34d41eab473cf3d01946cc5b3c897 Mon Sep 17 00:00:00 2001
From: pratiksha badheka <pratikshabadheka27@gmail.com>
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
