fix link style tests

uzhastik · Mar 10, 2024 · 7472f4e · 7472f4e
1 parent 2bd6831
commit 7472f4e
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/tests/providers/yandex/links/test_yq.py b/tests/providers/yandex/links/test_yq.py
@@ -18,21 +18,16 @@
 
 from unittest import mock
 
-from airflow.models.xcom import XCom
 from airflow.models.taskinstance import TaskInstance
-
+from airflow.models.xcom import XCom
 from airflow.providers.yandex.links.yq import YQLink
 from tests.test_utils.mock_operators import MockOperator
 
 
 def test_persist():
     mock_context = mock.MagicMock()
 
-    YQLink.persist(
-        context=mock_context,
-        task_instance=MockOperator(task_id="test_task_id"),
-        web_link="g.com"
-    )
+    YQLink.persist(context=mock_context, task_instance=MockOperator(task_id="test_task_id"), web_link="g.com")
 
     ti = mock_context["ti"]
     ti.xcom_push.assert_called_once_with(