From 666bb067ae19ee221ee9a4437f5a2c0bc749b3a6 Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Fri, 4 Mar 2022 09:57:03 +0100 Subject: [PATCH] blackkky --- pyinfra/test/unit_tests/conftest.py | 1 + pyinfra/test/unit_tests/core_test.py | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/pyinfra/test/unit_tests/conftest.py b/pyinfra/test/unit_tests/conftest.py index 53202c7..7fc01f4 100644 --- a/pyinfra/test/unit_tests/conftest.py +++ b/pyinfra/test/unit_tests/conftest.py @@ -33,4 +33,5 @@ def mock_payload(): def mock_make_load_data(): def load_data(payload): return storage_data + return load_data diff --git a/pyinfra/test/unit_tests/core_test.py b/pyinfra/test/unit_tests/core_test.py index 7e1a28b..f843ed8 100644 --- a/pyinfra/test/unit_tests/core_test.py +++ b/pyinfra/test/unit_tests/core_test.py @@ -17,7 +17,9 @@ def test_analyse_returns_analysis(mock_post, storage_data, mock_response): @patch("requests.post") -def test_process_returns_dossier_id_file_id_predictions(mock_post, mock_make_load_data, storage_data, mock_response, mock_payload): +def test_process_returns_dossier_id_file_id_predictions( + mock_post, mock_make_load_data, storage_data, mock_response, mock_payload +): mock_post.return_value = mock_response analyze = make_analyzer(CONFIG.mock_analysis_endpoint)