Skip to content

Commit cdfd59d

Browse files
fix test for windows
1 parent 9c393e6 commit cdfd59d

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

tests/v2/file_operations/test_split_operation_integration.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,10 @@ def test_pdf_should_extract_splits():
3131
response = client.enqueue_and_get_result(
3232
SplitResponse,
3333
split_input,
34-
SplitParameters(getenv("MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID"), close_file=False),
34+
SplitParameters(
35+
getenv("MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID"),
36+
close_file=False,
37+
),
3538
)
3639
assert response.inference.file.page_count == 2
3740

@@ -53,7 +56,11 @@ def test_pdf_should_extract_splits():
5356
extracted_pdf.save_to_file(OUTPUT_DIR / f"split_{i + 1:03d}.pdf")
5457
for i in range(len(extracted_pdfs)):
5558
local_input = PathInput(OUTPUT_DIR / f"split_{i + 1:03d}.pdf")
56-
assert local_input.page_count == extracted_pdfs[i].get_page_count()
59+
try:
60+
assert local_input.page_count == extracted_pdfs[i].get_page_count()
61+
finally:
62+
local_input.close()
63+
split_input.close()
5764

5865

5966
@pytest.fixture(scope="module", autouse=True)

0 commit comments

Comments
 (0)