From 1a3671cf97aef4d799726d6149f37e3764f672e1 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Fri, 25 Aug 2023 11:41:35 +0100 Subject: [PATCH] ruff --- .../tests/app/service/test_hub_service.py | 30 +++++++++---------- 1 file changed, 14 insertions(+), 16 deletions(-) diff --git a/openbb_sdk/sdk/core/tests/app/service/test_hub_service.py b/openbb_sdk/sdk/core/tests/app/service/test_hub_service.py index b3ce7e9b726..1457b1adf99 100644 --- a/openbb_sdk/sdk/core/tests/app/service/test_hub_service.py +++ b/openbb_sdk/sdk/core/tests/app/service/test_hub_service.py @@ -25,17 +25,16 @@ def test_connect_with_email_password(): mock_hub_session = MagicMock(spec=HubSession) with patch( "requests.post", return_value=MagicMock(status_code=200, json=lambda: {}) + ), patch.object( + HubService, + "_get_session_from_email_password", + return_value=mock_hub_session, ): - with patch.object( - HubService, - "_get_session_from_email_password", - return_value=mock_hub_session, - ): - hub_service = HubService() - result = hub_service.connect(email="test@example.com", password="password") + hub_service = HubService() + result = hub_service.connect(email="test@example.com", password="password") - assert result == mock_hub_session - assert hub_service.session == mock_hub_session + assert result == mock_hub_session + assert hub_service.session == mock_hub_session def test_connect_with_sdk_token(): @@ -43,15 +42,14 @@ def test_connect_with_sdk_token(): mock_hub_session = MagicMock(spec=HubSession) with patch( "requests.post", return_value=MagicMock(status_code=200, json=lambda: {}) + ), patch.object( + HubService, "_get_session_from_sdk_token", return_value=mock_hub_session ): - with patch.object( - HubService, "_get_session_from_sdk_token", return_value=mock_hub_session - ): - hub_service = HubService() - result = hub_service.connect(pat="pat") + hub_service = HubService() + result = hub_service.connect(pat="pat") - assert result == mock_hub_session - assert hub_service.session == mock_hub_session + assert result == mock_hub_session + assert hub_service.session == mock_hub_session def test_connect_without_credentials():