Fix example import tests after move of providers to new structure (#4…

…6217)
apache · Jan 28, 2025 · 085084b · 085084b
1 parent 4db60c1
commit 085084b
Show file tree

Hide file tree

Showing 12 changed files with 41 additions and 30 deletions.
diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam.py b/providers/apache/beam/tests/system/apache/beam/example_beam.py
@@ -21,14 +21,15 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     GCS_INPUT,
     GCS_JAR_DIRECT_RUNNER_BUCKET_NAME,
     GCS_JAR_DIRECT_RUNNER_OBJECT_NAME,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
 from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py b/providers/apache/beam/tests/system/apache/beam/example_beam_java_flink.py
@@ -21,14 +21,15 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     GCS_INPUT,
     GCS_JAR_FLINK_RUNNER_BUCKET_NAME,
     GCS_JAR_FLINK_RUNNER_OBJECT_NAME,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
 from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py b/providers/apache/beam/tests/system/apache/beam/example_beam_java_spark.py
@@ -21,14 +21,15 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     GCS_INPUT,
     GCS_JAR_SPARK_RUNNER_BUCKET_NAME,
     GCS_JAR_SPARK_RUNNER_OBJECT_NAME,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
 from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_go.py b/providers/apache/beam/tests/system/apache/beam/example_go.py
@@ -21,9 +21,7 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     DEFAULT_ARGS,
     GCP_PROJECT_ID,
     GCS_GO,
@@ -32,6 +30,9 @@
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
 from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_go_dataflow.py
@@ -21,9 +21,7 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     DEFAULT_ARGS,
     GCP_PROJECT_ID,
     GCS_GO_DATAFLOW_ASYNC,
@@ -32,6 +30,9 @@
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunGoPipelineOperator
 from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus
 from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
 from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor

diff --git a/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_java_dataflow.py
@@ -21,16 +21,17 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     GCS_JAR_DATAFLOW_RUNNER_BUCKET_NAME,
     GCS_JAR_DATAFLOW_RUNNER_OBJECT_NAME,
     GCS_OUTPUT,
     GCS_STAGING,
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunJavaPipelineOperator
 from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_python.py b/providers/apache/beam/tests/system/apache/beam/example_python.py
@@ -21,9 +21,7 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     DEFAULT_ARGS,
     GCP_PROJECT_ID,
     GCS_OUTPUT,
@@ -32,6 +30,9 @@
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
 from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_python_async.py b/providers/apache/beam/tests/system/apache/beam/example_python_async.py
@@ -21,9 +21,7 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     DEFAULT_ARGS,
     GCP_PROJECT_ID,
     GCS_OUTPUT,
@@ -32,6 +30,9 @@
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
 from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
 
 with models.DAG(

diff --git a/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py b/providers/apache/beam/tests/system/apache/beam/example_python_dataflow.py
@@ -21,9 +21,7 @@
 
 from __future__ import annotations
 
-from airflow import models
-from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
-from airflow.providers.apache.beam.tests.system.apache.beam.utils import (
+from providers.apache.beam.tests.system.apache.beam.utils import (
     DEFAULT_ARGS,
     GCP_PROJECT_ID,
     GCS_OUTPUT,
@@ -32,6 +30,9 @@
     GCS_TMP,
     START_DATE,
 )
+
+from airflow import models
+from airflow.providers.apache.beam.operators.beam import BeamRunPythonPipelineOperator
 from airflow.providers.google.cloud.hooks.dataflow import DataflowJobStatus
 from airflow.providers.google.cloud.operators.dataflow import DataflowConfiguration
 from airflow.providers.google.cloud.sensors.dataflow import DataflowJobStatusSensor

diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage.py b/providers/openlineage/tests/system/openlineage/example_openlineage.py
@@ -19,8 +19,9 @@
 import os
 from datetime import datetime
 
+from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
+
 from airflow import DAG
-from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
 from airflow.providers.standard.operators.python import PythonOperator
 
 

diff --git a/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py b/providers/openlineage/tests/system/openlineage/example_openlineage_mapped_sensor.py
@@ -19,9 +19,10 @@
 import os
 from datetime import datetime, timedelta
 
+from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
+
 from airflow import DAG
 from airflow.models import Variable
-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
 

diff --git a/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py b/providers/tests/system/google/cloud/gcs/example_gcs_upload_download.py
@@ -29,8 +29,8 @@
 from airflow.providers.google.cloud.operators.gcs import GCSCreateBucketOperator, GCSDeleteBucketOperator
 from airflow.providers.google.cloud.transfers.gcs_to_local import GCSToLocalFilesystemOperator
 from airflow.providers.google.cloud.transfers.local_to_gcs import LocalFilesystemToGCSOperator
-from airflow.providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
 from airflow.utils.trigger_rule import TriggerRule
+from providers.openlineage.tests.system.openlineage.operator import OpenLineageTestOperator
 
 from providers.tests.system.google import DEFAULT_GCP_SYSTEM_TEST_PROJECT_ID