Skip to content

Commit 82a03f3

Browse files
fix tests
1 parent 4144d49 commit 82a03f3

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

tests/v2/file_operations/test_crop_operation_integration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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)

tests/v2/product/classification/test_classification_integration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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"

0 commit comments

Comments
 (0)