Skip to content

Commit

Permalink
fix ruff format
Browse files Browse the repository at this point in the history
  • Loading branch information
gopidesupavan committed Nov 5, 2024
1 parent 6453ba5 commit c6eaf74
Showing 1 changed file with 19 additions and 16 deletions.
35 changes: 19 additions & 16 deletions providers/tests/standard/test_module_redirect_finder.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,22 +25,25 @@

@pytest.mark.skipif(not AIRFLOW_V_3_0_PLUS, reason="Valid for Airflow 3.0+ only")
class TestModuleRedirectFinder:
@pytest.mark.parametrize("old_module, new_module", [
("airflow.operators.python", "airflow.providers.standard.operators.python"),
("airflow.operators.bash", "airflow.providers.standard.operators.bash"),
("airflow.operators.datetime", "airflow.providers.standard.operators.datetime"),
("airflow.operators.weekday", "airflow.providers.standard.operators.weekday"),
("airflow.sensors.bash", "airflow.providers.standard.sensors.bash"),
("airflow.sensors.date_time", "airflow.providers.standard.sensors.date_time"),
("airflow.sensors.python", "airflow.providers.standard.sensors.python"),
("airflow.sensors.time", "airflow.providers.standard.sensors.time"),
("airflow.sensors.time_delta", "airflow.providers.standard.sensors.time_delta"),
("airflow.sensors.weekday", "airflow.providers.standard.sensors.weekday"),
("airflow.hooks.filesystem", "airflow.providers.standard.hooks.filesystem"),
("airflow.hooks.package_index", "airflow.providers.standard.hooks.package_index"),
("airflow.hooks.subprocess", "airflow.providers.standard.hooks.subprocess"),
("airflow.utils.python_virtualenv", "airflow.providers.standard.utils.python_virtualenv"),
])
@pytest.mark.parametrize(
"old_module, new_module",
[
("airflow.operators.python", "airflow.providers.standard.operators.python"),
("airflow.operators.bash", "airflow.providers.standard.operators.bash"),
("airflow.operators.datetime", "airflow.providers.standard.operators.datetime"),
("airflow.operators.weekday", "airflow.providers.standard.operators.weekday"),
("airflow.sensors.bash", "airflow.providers.standard.sensors.bash"),
("airflow.sensors.date_time", "airflow.providers.standard.sensors.date_time"),
("airflow.sensors.python", "airflow.providers.standard.sensors.python"),
("airflow.sensors.time", "airflow.providers.standard.sensors.time"),
("airflow.sensors.time_delta", "airflow.providers.standard.sensors.time_delta"),
("airflow.sensors.weekday", "airflow.providers.standard.sensors.weekday"),
("airflow.hooks.filesystem", "airflow.providers.standard.hooks.filesystem"),
("airflow.hooks.package_index", "airflow.providers.standard.hooks.package_index"),
("airflow.hooks.subprocess", "airflow.providers.standard.hooks.subprocess"),
("airflow.utils.python_virtualenv", "airflow.providers.standard.utils.python_virtualenv"),
],
)
def test_module_redirect_finder_for_old_location(self, old_module, new_module):
try:
assert importlib.import_module(old_module).__name__ == new_module
Expand Down

0 comments on commit c6eaf74

Please sign in to comment.