File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1212from mindee .input .path_input import PathInput
1313from mindee .v2 .client import Client
1414from 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
1818def 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 )
You can’t perform that action at this time.
0 commit comments