refactor(AIR311): reorg test cases

Author: Lee-W

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

@@ -9,11 +9,17 @@
     expand_alias_to_datasets,
 )
 from airflow.datasets.metadata import Metadata
+from airflow.decorators import dag, setup, task, task_group, teardown
+from airflow.io.path import ObjectStoragePath
+from airflow.io.storage import attach
+from airflow.models import DAG as DAGFromModel
 from airflow.models.baseoperator import chain, chain_linear, cross_downstream
 from airflow.models.baseoperatorlink import BaseOperatorLink
+from airflow.models.dag import DAG as DAGFromDag
 from airflow.timetables.datasets import DatasetOrTimeSchedule
 from airflow.utils.dag_parsing_context import get_parsing_context
 
+# airflow
 DatasetFromRoot()
 
 # airflow.datasets
@@ -24,6 +30,20 @@
 Metadata()
 expand_alias_to_datasets()
 
+# airflow.decorators
+dag()
+task()
+task_group()
+setup()
+teardown()
+
+# airflow.io
+ObjectStoragePath()
+attach()
+
+# airflow.models
+DAGFromModel()
+
 # airflow.models.baseoperator
 chain()
 chain_linear()
@@ -32,29 +52,10 @@
 # airflow.models.baseoperatolinker
 BaseOperatorLink()
 
+# airflow.models.dag
+DAGFromDag()
 # airflow.timetables.datasets
 DatasetOrTimeSchedule()
 
 # airflow.utils.dag_parsing_context
 get_parsing_context()
-
-
-from airflow.decorators import dag, task, task_group, setup, teardown
-
-dag()
-task()
-task_group()
-setup()
-teardown()
-
-from airflow.models.dag import DAG as DAGFromDag
-from airflow.models import DAG as DAGFromModel
-
-DAGFromDag()
-DAGFromModel()
-
-from airflow.io.path import ObjectStoragePath
-from airflow.io.storage import attach
-
-ObjectStoragePath()
-attach()