test: reorg test cases

Author: Lee-W

crates/ruff_linter/resources/test/fixtures/airflow/AIR302_names.py

@@ -45,16 +45,6 @@
 from airflow.listeners.spec.dataset import on_dataset_changed, on_dataset_created
 from airflow.metrics.validators import AllowListValidator, BlockListValidator
 from airflow.models.baseoperator import chain, chain_linear, cross_downstream
-
-
-
-
-
-
-
-
-
-
 from airflow.operators.subdag import SubDagOperator
 from airflow.providers.amazon.aws.auth_manager.avp.entities import AvpEntities
 from airflow.providers.amazon.aws.datasets import s3
@@ -71,22 +61,6 @@
 from airflow.secrets.local_filesystem import LocalFilesystemBackend, load_connections
 from airflow.security.permissions import RESOURCE_DATASET
 from airflow.sensors.base_sensor_operator import BaseSensorOperator
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-from airflow.sensors.time_delta_sensor import TimeDeltaSensor
 from airflow.timetables.datasets import DatasetOrTimeSchedule
 from airflow.timetables.simple import DatasetTriggeredTimetable
 from airflow.triggers.external_task import TaskStateTrigger
@@ -230,21 +204,6 @@
 BaseSensorOperator()
 
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
 # airflow.timetables
 DatasetOrTimeSchedule()
 DatasetTriggeredTimetable()

crates/ruff_linter/resources/test/fixtures/airflow/AIR303.py

@@ -100,6 +100,12 @@
 from airflow.kubernetes.volume import Volume
 from airflow.kubernetes.volume_mount import VolumeMount
 from airflow.macros.hive import closest_ds_partition, max_partition
+from airflow.operators import (
+    dagrun_operator,
+    dummy_operator,
+    email_operator,
+    python_operator,
+)
 from airflow.operators.bash import BashOperator
 from airflow.operators.bash_operator import BashOperator as LegacyBashOperator
 from airflow.operators.check_operator import (
@@ -115,6 +121,9 @@
 from airflow.operators.datetime import BranchDateTimeOperator
 from airflow.operators.docker_operator import DockerOperator
 from airflow.operators.druid_check_operator import DruidCheckOperator
+from airflow.operators.dummy import DummyOperator, EmptyOperator
+from airflow.operators.email import EmailOperator
+from airflow.operators.email_operator import EmailOperator
 from airflow.operators.gcs_to_s3 import GCSToS3Operator
 from airflow.operators.google_api_to_s3_transfer import (
     GoogleApiToS3Operator,
@@ -153,6 +162,12 @@
     PrestoToMySqlOperator,
     PrestoToMySqlTransfer,
 )
+from airflow.operators.python import (
+    BranchPythonOperator,
+    PythonOperator,
+    PythonVirtualenvOperator,
+    ShortCircuitOperator,
+)
 from airflow.operators.redshift_to_s3_operator import (
     RedshiftToS3Operator,
     RedshiftToS3Transfer,
@@ -189,8 +204,14 @@
 from airflow.operators.sqlite_operator import SqliteOperator
 from airflow.operators.trigger_dagrun import TriggerDagRunOperator
 from airflow.operators.weekday import BranchDayOfWeekOperator
+from airflow.sensors import external_task_sensor
 from airflow.sensors.date_time import DateTimeSensor
-from airflow.sensors.external_task import ExternalTaskMarker, ExternalTaskSensor
+from airflow.sensors.date_time_sensor import DateTimeSensor
+from airflow.sensors.external_task import (
+    ExternalTaskMarker,
+    ExternalTaskSensor,
+    ExternalTaskSensorLink,
+)
 from airflow.sensors.filesystem import FileSensor
 from airflow.sensors.hive_partition_sensor import HivePartitionSensor
 from airflow.sensors.http_sensor import HttpSensor
@@ -408,87 +429,43 @@
 # apache-airflow-providers-zendesk
 ZendeskHook()
 
+# apache-airflow-providers-smtp
+EmailOperator()
+email_operator.EmailOperator()
+
 # apache-airflow-providers-standard
-FileSensor()
-TriggerDagRunOperator()
-ExternalTaskMarker(), ExternalTaskSensor()
 BranchDateTimeOperator()
 BranchDayOfWeekOperator()
+BranchPythonOperator()
 DateTimeSensor()
-TimeSensor()
-TimeDeltaSensor()
+DateTimeTrigger()
 DayOfWeekSensor()
+DummyOperator()
+EmptyOperator()
+ExternalTaskMarker()
+ExternalTaskSensor()
+ExternalTaskSensorLink()
+FileSensor()
+FileTrigger()
 FSHook()
 PackageIndexHook()
 SubprocessHook()
+ShortCircuitOperator()
+TimeDeltaSensor()
+TimeSensor()
+TriggerDagRunOperator()
 WorkflowTrigger()
-FileTrigger()
-DateTimeTrigger()
-
-from airflow.operators import dummy_operator
-from airflow.operators.dummy import DummyOperator, EmptyOperator
-
-DummyOperator()
-EmptyOperator()
-dummy_operator.DummyOperator()
-dummy_operator.EmptyOperator()
-
-from airflow.operators import email_operator
-from airflow.operators.email import EmailOperator
-
-EmailOperator()
-email_operator.EmailOperator()
-
-from airflow.operators import dagrun_operator, dummy_operator, python_operator
-from airflow.operators.dummy import DummyOperator, EmptyOperator
-from airflow.operators.email_operator import EmailOperator
-from airflow.operators.python import (
-    BranchPythonOperator,
-    PythonOperator,
-    PythonVirtualenvOperator,
-    ShortCircuitOperator,
-)
-from airflow.sensors.date_time_sensor import DateTimeSensor
-from airflow.sensors.external_task import (
-    ExternalTaskMarker,
-    ExternalTaskSensor,
-    ExternalTaskSensorLink,
-)
-from airflow.sensors.external_task_sensor import (
-    ExternalTaskMarker as ExternalTaskMarkerFromExternalTaskSensor,
-)
-from airflow.sensors.external_task_sensor import (
-    ExternalTaskSensor as ExternalTaskSensorFromExternalTaskSensor,
-)
-from airflow.sensors.external_task_sensor import (
-    ExternalTaskSensorLink as ExternalTaskSensorLinkFromExternalTaskSensor,
-)
+PythonOperator()
+PythonVirtualenvOperator()
 
 dagrun_operator.TriggerDagRunLink()
 dagrun_operator.TriggerDagRunOperator()
-
-# airflow.sensors.date_time_sensor
-DateTimeSensor()
-
-# airflow.sensors.external_task
-ExternalTaskSensorLink()
-ExternalTaskMarker()
-ExternalTaskSensor()
-
-# airflow.sensors.external_task_sensor
-ExternalTaskMarkerFromExternalTaskSensor()
-ExternalTaskSensorFromExternalTaskSensor()
-ExternalTaskSensorLinkFromExternalTaskSensor()
-
-# airflow.sensors.time_delta_sensor
-TimeDeltaSensor()
-
-# airflow.operators
+dummy_operator.DummyOperator()
+dummy_operator.EmptyOperator()
+external_task_sensor.ExternalTaskSensorLink()
+external_task_sensor.ExternalTaskMarker()
+external_task_sensor.ExternalTaskSensor()
 python_operator.BranchPythonOperator()
 python_operator.PythonOperator()
 python_operator.PythonVirtualenvOperator()
 python_operator.ShortCircuitOperator()
-BranchPythonOperator()
-PythonOperator()
-PythonVirtualenvOperator()
-ShortCircuitOperator()