diff --git a/tests/cli/commands/test_pool_command.py b/tests/cli/commands/test_pool_command.py index 96809fb4df712..ec175912d8503 100644 --- a/tests/cli/commands/test_pool_command.py +++ b/tests/cli/commands/test_pool_command.py @@ -20,7 +20,6 @@ import io import json import os -import unittest from contextlib import redirect_stdout import pytest @@ -33,17 +32,14 @@ from airflow.utils.db import add_default_pool_if_not_exists -class TestCliPools(unittest.TestCase): +class TestCliPools: @classmethod - def setUpClass(cls): + def setup_class(cls): cls.dagbag = models.DagBag(include_examples=True) cls.parser = cli_parser.get_parser() - - def setUp(self): - super().setUp() settings.configure_orm() - self.session = Session - self._cleanup() + cls.session = Session + cls._cleanup() def tearDown(self): self._cleanup() diff --git a/tests/cli/commands/test_task_command.py b/tests/cli/commands/test_task_command.py index 7e10ad21a1c4d..5b9f8ace71919 100644 --- a/tests/cli/commands/test_task_command.py +++ b/tests/cli/commands/test_task_command.py @@ -33,7 +33,6 @@ import pendulum import pytest -from parameterized import parameterized from airflow import DAG from airflow.cli import cli_parser @@ -336,12 +335,13 @@ def test_cli_test_with_env_vars(self): assert "foo=bar" in output assert "AIRFLOW_TEST_MODE=True" in output - @parameterized.expand( + @pytest.mark.parametrize( + "option", [ - ("--ignore-all-dependencies",), - ("--ignore-depends-on-past",), - ("--ignore-dependencies",), - ("--force",), + "--ignore-all-dependencies", + "--ignore-depends-on-past", + "--ignore-dependencies", + "--force", ], ) def test_cli_run_invalid_raw_option(self, option: str):