update test syntax

Author: sebastianMindee

tests/v2/test_client.py

@@ -124,9 +124,7 @@ def test_enqueue_path_with_env_token(custom_base_url_client):
         f"{FILE_TYPES_DIR}/receipt.jpg"
     )
     with pytest.raises(MindeeHTTPErrorV2):
-        custom_base_url_client.enqueue_inference(
-            input_doc, InferenceParameters("dummy-model")
-        )
+        custom_base_url_client.enqueue(input_doc, InferenceParameters("dummy-model"))
 
 
 @pytest.mark.v2
@@ -135,7 +133,8 @@ def test_enqueue_and_parse_path_with_env_token(custom_base_url_client):
         f"{FILE_TYPES_DIR}/receipt.jpg"
     )
     with pytest.raises(MindeeHTTPErrorV2):
-        custom_base_url_client.enqueue_and_get_inference(
+        custom_base_url_client.enqueue_and_get_result(
+            InferenceResponse,
             input_doc,
             InferenceParameters(
                 "dummy-model",
@@ -172,16 +171,16 @@ def test_loads_from_prediction():
 
 @pytest.mark.v2
 def test_get_inference(custom_base_url_client):
-    response = custom_base_url_client.get_inference(
-        "12345678-1234-1234-1234-123456789ABC"
+    response = custom_base_url_client.get_result(
+        InferenceResponse, "12345678-1234-1234-1234-123456789ABC"
     )
     _assert_findoc_inference(response)
 
 
 @pytest.mark.v2
 def test_error_handling(custom_base_url_client):
     with pytest.raises(MindeeHTTPErrorV2) as e:
-        custom_base_url_client.enqueue_inference(
+        custom_base_url_client.enqueue(
             PathInput(
                 V2_DATA_DIR / "products" / "financial_document" / "default_sample.jpg"
             ),

tests/v2/test_client_integration.py

@@ -58,8 +58,8 @@ def test_parse_file_empty_multiple_pages_must_succeed(
         alias="py_integration_empty_multiple",
     )
 
-    response: InferenceResponse = v2_client.enqueue_and_get_inference(
-        input_source, params
+    response: InferenceResponse = v2_client.enqueue_and_get_result(
+        InferenceResponse, input_source, params
     )
     _basic_assert_success(response=response, page_count=2, model_id=findoc_model_id)
 
@@ -99,8 +99,8 @@ def test_parse_file_empty_single_page_options_must_succeed(
         confidence=True,
         alias="py_integration_empty_page_options",
     )
-    response: InferenceResponse = v2_client.enqueue_and_get_inference(
-        input_source, params
+    response: InferenceResponse = v2_client.enqueue_and_get_result(
+        InferenceResponse, input_source, params
     )
     _basic_assert_success(response=response, page_count=1, model_id=findoc_model_id)
 
@@ -137,8 +137,8 @@ def test_parse_file_filled_single_page_must_succeed(
         text_context="this is an invoice.",
     )
 
-    response: InferenceResponse = v2_client.enqueue_and_get_inference(
-        input_source, params
+    response: InferenceResponse = v2_client.enqueue_and_get_result(
+        InferenceResponse, input_source, params
     )
     _basic_assert_success(response=response, page_count=1, model_id=findoc_model_id)
 
@@ -178,7 +178,7 @@ def test_invalid_uuid_must_throw_error(v2_client: ClientV2) -> None:
     )
 
     with pytest.raises(MindeeHTTPErrorV2) as exc_info:
-        v2_client.enqueue_inference(input_source, params)
+        v2_client.enqueue(input_source, params)
 
     exc: MindeeHTTPErrorV2 = exc_info.value
     assert exc.status == 422
@@ -199,7 +199,7 @@ def test_unknown_model_must_throw_error(v2_client: ClientV2) -> None:
     params = InferenceParameters(model_id="fc405e37-4ba4-4d03-aeba-533a8d1f0f21")
 
     with pytest.raises(MindeeHTTPErrorV2) as exc_info:
-        v2_client.enqueue_inference(input_source, params)
+        v2_client.enqueue(input_source, params)
 
     exc: MindeeHTTPErrorV2 = exc_info.value
     assert exc.status == 404
@@ -232,7 +232,7 @@ def test_unknown_webhook_ids_must_throw_error(
     )
 
     with pytest.raises(MindeeHTTPErrorV2) as exc_info:
-        v2_client.enqueue_inference(input_source, params)
+        v2_client.enqueue(input_source, params)
 
     exc: MindeeHTTPErrorV2 = exc_info.value
     assert exc.status == 422
@@ -263,8 +263,8 @@ def test_blank_url_input_source_must_succeed(
         webhook_ids=[],
         alias="py_integration_url_source",
     )
-    response: InferenceResponse = v2_client.enqueue_and_get_inference(
-        input_source, params
+    response: InferenceResponse = v2_client.enqueue_and_get_result(
+        InferenceResponse, input_source, params
     )
     _basic_assert_success(response=response, page_count=1, model_id=findoc_model_id)
 
@@ -294,8 +294,8 @@ def test_data_schema_must_succeed(
         data_schema=data_schema_replace_path.read_text(),
         alias="py_integration_data_schema_replace",
     )
-    response: InferenceResponse = v2_client.enqueue_and_get_inference(
-        input_source, params
+    response: InferenceResponse = v2_client.enqueue_and_get_result(
+        InferenceResponse, input_source, params
     )
     _basic_assert_success(response=response, page_count=1, model_id=findoc_model_id)
     assert response.inference.active_options.data_schema.replace is True