diff --git a/tests/decorators/test_branch_external_python.py b/tests/decorators/test_branch_external_python.py index ca62e038f1ea5..745edafae86ce 100644 --- a/tests/decorators/test_branch_external_python.py +++ b/tests/decorators/test_branch_external_python.py @@ -27,7 +27,11 @@ pytestmark = pytest.mark.db_test -class Test_BranchPythonDecoratedOperator: +class Test_BranchExternalPythonDecoratedOperator: + # when run in "Parallel" test run environment, sometimes this test runs for a long time + # because creating virtualenv and starting new Python interpreter creates a lot of IO/contention + # possibilities. So we are increasing the timeout for this test to 3x of the default timeout + @pytest.mark.execution_timeout(180) @pytest.mark.parametrize("branch_task_name", ["task_1", "task_2"]) def test_branch_one(self, dag_maker, branch_task_name): @task diff --git a/tests/decorators/test_branch_virtualenv.py b/tests/decorators/test_branch_virtualenv.py index 66f539ca4c988..2b5f9bb95ed73 100644 --- a/tests/decorators/test_branch_virtualenv.py +++ b/tests/decorators/test_branch_virtualenv.py @@ -25,7 +25,11 @@ pytestmark = pytest.mark.db_test -class Test_BranchPythonDecoratedOperator: +class Test_BranchPythonVirtualenvDecoratedOperator: + # when run in "Parallel" test run environment, sometimes this test runs for a long time + # because creating virtualenv and starting new Python interpreter creates a lot of IO/contention + # possibilities. So we are increasing the timeout for this test to 3x of the default timeout + @pytest.mark.execution_timeout(180) @pytest.mark.parametrize("branch_task_name", ["task_1", "task_2"]) def test_branch_one(self, dag_maker, branch_task_name): @task