1010from linkup import (
1111 LinkupAuthenticationError ,
1212 LinkupClient ,
13- LinkupInvalidRequestError ,
14- LinkupSearchResults ,
15- LinkupSource ,
16- LinkupSourcedAnswer ,
17- LinkupUnknownError ,
18- )
19- from linkup .errors import (
2013 LinkupFailedFetchError ,
14+ LinkupFetchResponse ,
2115 LinkupInsufficientCreditError ,
16+ LinkupInvalidRequestError ,
2217 LinkupNoResultError ,
23- LinkupTooManyRequestsError ,
24- )
25- from linkup .types import (
26- LinkupFetchResponse ,
2718 LinkupSearchImageResult ,
19+ LinkupSearchResults ,
2820 LinkupSearchStructuredResponse ,
2921 LinkupSearchTextResult ,
22+ LinkupSource ,
23+ LinkupSourcedAnswer ,
24+ LinkupTooManyRequestsError ,
25+ LinkupUnknownError ,
3026)
3127
3228
@@ -306,9 +302,9 @@ def test_search(
306302 mock_request_response_content : bytes ,
307303 expected_search_response : Any ,
308304) -> None :
309- mocker .patch ("linkup.client .date" ).today .return_value = date (2000 , 1 , 1 )
305+ mocker .patch ("linkup._client .date" ).today .return_value = date (2000 , 1 , 1 )
310306 request_mock = mocker .patch (
311- "linkup.client .LinkupClient._request" ,
307+ "linkup._client .LinkupClient._request" ,
312308 return_value = Response (
313309 status_code = 200 ,
314310 content = mock_request_response_content ,
@@ -339,9 +335,9 @@ async def test_async_search(
339335 mock_request_response_content : bytes ,
340336 expected_search_response : Any ,
341337) -> None :
342- mocker .patch ("linkup.client .date" ).today .return_value = date (2000 , 1 , 1 )
338+ mocker .patch ("linkup._client .date" ).today .return_value = date (2000 , 1 , 1 )
343339 request_mock = mocker .patch (
344- "linkup.client .LinkupClient._async_request" ,
340+ "linkup._client .LinkupClient._async_request" ,
345341 return_value = Response (
346342 status_code = 200 ,
347343 content = mock_request_response_content ,
@@ -483,7 +479,7 @@ def test_search_error(
483479 expected_exception : Any ,
484480) -> None :
485481 request_mock = mocker .patch (
486- "linkup.client .LinkupClient._request" ,
482+ "linkup._client .LinkupClient._request" ,
487483 return_value = Response (
488484 status_code = mock_request_response_status_code ,
489485 content = mock_request_response_content ,
@@ -508,7 +504,7 @@ async def test_async_search_error(
508504 expected_exception : Any ,
509505) -> None :
510506 request_mock = mocker .patch (
511- "linkup.client .LinkupClient._async_request" ,
507+ "linkup._client .LinkupClient._async_request" ,
512508 return_value = Response (
513509 status_code = mock_request_response_status_code ,
514510 content = mock_request_response_content ,
@@ -549,7 +545,7 @@ def test_fetch(
549545 expected_fetch_response : Any ,
550546) -> None :
551547 request_mock = mocker .patch (
552- "linkup.client .LinkupClient._request" ,
548+ "linkup._client .LinkupClient._request" ,
553549 return_value = Response (
554550 status_code = 200 ,
555551 content = mock_request_response_content ,
@@ -580,7 +576,7 @@ async def test_async_fetch(
580576 expected_fetch_response : Any ,
581577) -> None :
582578 request_mock = mocker .patch (
583- "linkup.client .LinkupClient._async_request" ,
579+ "linkup._client .LinkupClient._async_request" ,
584580 return_value = Response (
585581 status_code = 200 ,
586582 content = mock_request_response_content ,
@@ -644,7 +640,7 @@ def test_fetch_error(
644640 expected_exception : Any ,
645641) -> None :
646642 request_mock = mocker .patch (
647- "linkup.client .LinkupClient._request" ,
643+ "linkup._client .LinkupClient._request" ,
648644 return_value = Response (
649645 status_code = mock_request_response_status_code ,
650646 content = mock_request_response_content ,
@@ -669,7 +665,7 @@ async def test_async_fetch_error(
669665 expected_exception : Any ,
670666) -> None :
671667 request_mock = mocker .patch (
672- "linkup.client .LinkupClient._async_request" ,
668+ "linkup._client .LinkupClient._async_request" ,
673669 return_value = Response (
674670 status_code = mock_request_response_status_code ,
675671 content = mock_request_response_content ,
0 commit comments