Skip to content

Commit cdaef22

Browse files
committed
check options in inference
1 parent d15d4f2 commit cdaef22

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

tests/test_client_v2_integration.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def test_parse_file_empty_multiple_pages_must_succeed(
4646
polygon=False,
4747
confidence=False,
4848
webhook_ids=[],
49-
alias='py_integration_empty_multiple'
49+
alias="py_integration_empty_multiple",
5050
)
5151

5252
response: InferenceResponse = v2_client.enqueue_and_get_inference(
@@ -81,7 +81,7 @@ def test_parse_file_filled_single_page_must_succeed(
8181
polygon=False,
8282
confidence=False,
8383
webhook_ids=[],
84-
alias='py_integration_empty_multipage'
84+
alias="py_integration_empty_multipage",
8585
)
8686

8787
response: InferenceResponse = v2_client.enqueue_and_get_inference(
@@ -95,6 +95,12 @@ def test_parse_file_filled_single_page_must_succeed(
9595
assert response.inference.file.name == "default_sample.jpg"
9696
assert response.inference.file.page_count == 1
9797

98+
assert response.inference.active_options is not None
99+
assert response.inference.active_options.rag is False
100+
assert response.inference.active_options.raw_text is False
101+
assert response.inference.active_options.polygon is False
102+
assert response.inference.active_options.confidence is False
103+
98104
assert response.inference.model is not None
99105
assert response.inference.model.id == findoc_model_id
100106

@@ -141,7 +147,7 @@ def test_url_input_source_must_not_raise_errors(
141147
polygon=False,
142148
confidence=False,
143149
webhook_ids=[],
144-
alias='py_integration_url_source'
150+
alias="py_integration_url_source",
145151
)
146152
response: InferenceResponse = v2_client.enqueue_and_get_inference(
147153
input_source, params

0 commit comments

Comments
 (0)