From 8e157a52c965a07c2b0f3a03d18e6ccba18d2ba3 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 26 Sep 2023 14:46:24 +0000 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20[pre-commit.ci]=20Auto=20format?= =?UTF-8?q?=20from=20pre-commit.com=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cosmos/operators/local.py | 5 +---- tests/operators/test_local.py | 4 +++- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py index b3c7af63a..87e2b245c 100644 --- a/cosmos/operators/local.py +++ b/cosmos/operators/local.py @@ -40,10 +40,7 @@ FullOutputSubprocessHook, FullOutputSubprocessResult, ) -from cosmos.dbt.parser.output import ( - extract_log_issues, - parse_output -) +from cosmos.dbt.parser.output import extract_log_issues logger = get_logger(__name__) diff --git a/tests/operators/test_local.py b/tests/operators/test_local.py index a05509ee8..36462c0b6 100644 --- a/tests/operators/test_local.py +++ b/tests/operators/test_local.py @@ -303,6 +303,7 @@ def test_operator_execute_without_flags(mock_build_and_run_cmd, operator_class): task.execute(context={}) mock_build_and_run_cmd.assert_called_once_with(context={}) + @pytest.mark.integration def test_dbt_test_local_operator_on_warning_callback(): on_warning_callback_mock = MagicMock() @@ -314,13 +315,14 @@ def test_dbt_test_local_operator_on_warning_callback(): task_id="my-task", install_deps=True, dbt_cmd_flags=["--models", "stg_customers"], - on_warning_callback=on_warning_callback_mock + on_warning_callback=on_warning_callback_mock, ) test_operator run_test_dag(dag) on_warning_callback_mock.assert_called_once() + @pytest.mark.integration def test_on_warning_callback_not_triggered(): on_warning_callback_mock = MagicMock()