Skip to content

Commit 96f3b98

Browse files
committed
test: updated tests
1 parent b06af78 commit 96f3b98

File tree

1 file changed

+10
-33
lines changed

1 file changed

+10
-33
lines changed

_test_unstructured_client/unit/test_split_pdf_hook.py

Lines changed: 10 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -53,29 +53,6 @@ async def example():
5353
assert hook.api_successful_responses.get(operation_id) is None
5454

5555

56-
def test_unit_prepare_request_payload():
57-
"""Test prepare request payload method properly sets split_pdf_page to 'false'
58-
and removes files key."""
59-
test_form_data = {
60-
"files": ("test_file.pdf", b"test_file_content"),
61-
"split_pdf_page": "true",
62-
"parameter_1": "value_1",
63-
"parameter_2": "value_2",
64-
"parameter_3": "value_3",
65-
}
66-
expected_form_data = {
67-
"split_pdf_page": "false",
68-
"parameter_1": "value_1",
69-
"parameter_2": "value_2",
70-
"parameter_3": "value_3",
71-
}
72-
73-
payload = request_utils.prepare_request_payload(test_form_data)
74-
75-
assert payload != test_form_data
76-
assert payload, expected_form_data
77-
78-
7956
def test_unit_prepare_request_headers():
8057
"""Test prepare request headers method properly removes Content-Type and Content-Length headers."""
8158
test_headers = {
@@ -234,9 +211,9 @@ def test_unit_is_pdf_valid_pdf():
234211
file_name=filename,
235212
)
236213

237-
result = pdf_utils.is_pdf(file)
214+
result = pdf_utils.read_pdf(file)
238215

239-
assert result is True
216+
assert result is not None
240217

241218

242219
def test_unit_is_pdf_valid_pdf_without_file_extension():
@@ -249,36 +226,36 @@ def test_unit_is_pdf_valid_pdf_without_file_extension():
249226
file_name="uuid1234",
250227
)
251228

252-
result = pdf_utils.is_pdf(file)
229+
result = pdf_utils.read_pdf(file)
253230

254-
assert result is True
231+
assert result is not None
255232

256233

257234
def test_unit_is_pdf_invalid_extension():
258235
"""Test is pdf method returns False for file with invalid extension."""
259236
file = shared.Files(content=b"txt_content", file_name="test_file.txt")
260237

261-
result = pdf_utils.is_pdf(file)
238+
result = pdf_utils.read_pdf(file)
262239

263-
assert result is False
240+
assert result is None
264241

265242

266243
def test_unit_is_pdf_invalid_pdf():
267244
"""Test is pdf method returns False for file with invalid pdf content."""
268245
file = shared.Files(content=b"invalid_pdf_content", file_name="test_file.pdf")
269246

270-
result = pdf_utils.is_pdf(file)
247+
result = pdf_utils.read_pdf(file)
271248

272-
assert result is False
249+
assert result is None
273250

274251

275252
def test_unit_is_pdf_invalid_pdf_without_file_extension():
276253
"""Test is pdf method returns False for file with invalid pdf content without basing on file extension."""
277254
file = shared.Files(content=b"invalid_pdf_content", file_name="uuid1234")
278255

279-
result = pdf_utils.is_pdf(file)
256+
result = pdf_utils.read_pdf(file)
280257

281-
assert result is False
258+
assert result is not None
282259

283260

284261
def test_unit_get_starting_page_number_missing_key():

0 commit comments

Comments
 (0)