From 3c6dfed50845024b8231f2517369aaa42327072c Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 28 Mar 2022 19:22:31 +0200 Subject: [PATCH] made input size adjustable via test fixture --- test/unit_tests/conftest.py | 8 ++++---- test/unit_tests/image_extractor_test.py | 5 +++-- 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/test/unit_tests/conftest.py b/test/unit_tests/conftest.py index aa173cf..de5aa0a 100644 --- a/test/unit_tests/conftest.py +++ b/test/unit_tests/conftest.py @@ -100,9 +100,9 @@ def array_to_image(array): return Image.fromarray(np.uint8(array * 255), mode="RGB") -@pytest.fixture -def input_size(depth=3, width=10, height=15): - return width, height, depth +@pytest.fixture(params=[{'depth': 3}, {'width': 10}, {'height': 15}]) +def input_size(request): + return request.param.get("width", 10), request.param.get("height", 15), request.param.get("depth", 3) @pytest.fixture @@ -135,7 +135,7 @@ def metadata(images): def current_page_idx(): nonlocal page_idx - page_idx += random.randint(0, 2) + page_idx += random.randint(0, 3) return min(page_idx, len(images) - 1) def build_image_metadata(image): diff --git a/test/unit_tests/image_extractor_test.py b/test/unit_tests/image_extractor_test.py index 9ad2c9f..ffc311a 100644 --- a/test/unit_tests/image_extractor_test.py +++ b/test/unit_tests/image_extractor_test.py @@ -13,8 +13,9 @@ def test_image_extractor_mock(image_extractor, images): @pytest.mark.parametrize("extractor_type", ["parsable_pdf"]) -@pytest.mark.parametrize("batch_size", [0, 1, 2, 4, 8]) -def test_parsable_pdf_image_extractor(image_extractor, pdf, images, metadata): +@pytest.mark.parametrize("batch_size", [128]) +@pytest.mark.parametrize("input_size", [{"depth": 3, "width": 100, "height": 90}], indirect=["input_size"]) +def test_parsable_pdf_image_extractor(image_extractor, pdf, images, metadata, input_size): images_extracted, metadata_extracted = map(list, extract_images_from_pdf(pdf, image_extractor)) assert np.allclose(images_to_batch_tensor(images_extracted), images_to_batch_tensor(images)) assert list(metadata_extracted) == metadata