Skip to content

Commit ccfb58b

Browse files
fix other test
1 parent 0fad789 commit ccfb58b

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

tests/v2/file_operations/test_crop_operation_integration.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from mindee.input.path_input import PathInput
1313
from mindee.v2.client import Client
1414
from mindee.v2.file_operations.crop import extract_multiple_crops
15-
from tests.utils import OUTPUT_DIR, V2_PRODUCT_DATA_DIR
15+
from tests.utils import OUTPUT_DIR, V2_PRODUCT_DATA_DIR, cleanup_output_files
1616

1717

1818
def check_findoc_return(findoc_response: ExtractionResponse):
@@ -57,11 +57,11 @@ def test_image_should_extract_crops():
5757
extracted_crops.save_all_to_disk(OUTPUT_DIR)
5858
crop0_size = os.path.getsize(OUTPUT_DIR / output_files[0])
5959
crop1_size = os.path.getsize(OUTPUT_DIR / output_files[1])
60-
assert 170000 <= crop0_size <= 199685
61-
assert 180000 <= crop1_size <= 199433
60+
assert 100000 <= crop0_size <= 199685
61+
assert 100000 <= crop1_size <= 199433
6262

6363

64-
# @pytest.fixture(scope="module", autouse=True)
65-
# def cleanup():
66-
# yield
67-
# cleanup_output_files(output_files)
64+
@pytest.fixture(scope="module", autouse=True)
65+
def cleanup():
66+
yield
67+
cleanup_output_files(output_files)

0 commit comments

Comments
 (0)