File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -51,8 +51,8 @@ def test_image_should_extract_crops():
5151 )
5252 check_findoc_return (invoice_0 )
5353 extracted_images .save_all_to_disk (OUTPUT_DIR )
54- assert os .path .getsize (OUTPUT_DIR / "crop_001.jpg" ) == 198887
55- assert os .path .getsize (OUTPUT_DIR / "crop_002.jpg" ) == 197443
54+ assert os .path .getsize (OUTPUT_DIR / "crop_001.jpg" ) == 187601
55+ assert os .path .getsize (OUTPUT_DIR / "crop_002.jpg" ) == 197978
5656
5757
5858@pytest .fixture (scope = "module" , autouse = True )
Original file line number Diff line number Diff line change @@ -24,14 +24,14 @@ def test_classification_default_sample(
2424 v2_client : ClientV2 , classification_model_id : str
2525):
2626 input_source = PathInput (
27- V2_PRODUCT_DATA_DIR / "classification" / "default_invoice .jpg"
27+ V2_PRODUCT_DATA_DIR / "classification" / "default_sample .jpg"
2828 )
2929 response = v2_client .enqueue_and_get_result (
3030 ClassificationResponse ,
3131 input_source ,
3232 ClassificationParameters (classification_model_id ),
3333 )
3434 assert response .inference is not None
35- assert response .inference .file .name == "default_invoice .jpg"
35+ assert response .inference .file .name == "default_sample .jpg"
3636 assert response .inference .result .classification
3737 assert response .inference .result .classification .document_type == "invoice"
You can’t perform that action at this time.
0 commit comments