Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

test parametrization changed

Browse Source
This commit is contained in:
Matthias Bisping 2022-03-28 19:52:24 +02:00
parent 3c6dfed508
commit d44622dddc
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/unit_tests/conftest.py
View File

@ -100,9 +100,9 @@ def array_to_image(array):
return Image.fromarray(np.uint8(array * 255), mode="RGB") return Image.fromarray(np.uint8(array * 255), mode="RGB")
@pytest.fixture(params=[{'depth': 3}, {'width': 10}, {'height': 15}]) @pytest.fixture(params=[{"width": 10, "height": 15, "depth": 3}, {"width": 150, "height": 100, "depth": 3}])
def input_size(request): def input_size(request):
return request.param.get("width", 10), request.param.get("height", 15), request.param.get("depth", 3) return itemgetter("width", "height", "depth")(request.param)
@pytest.fixture @pytest.fixture

4
test/unit_tests/image_extractor_test.py
View File

@ -13,8 +13,8 @@ def test_image_extractor_mock(image_extractor, images):
@pytest.mark.parametrize("extractor_type", ["parsable_pdf"]) @pytest.mark.parametrize("extractor_type", ["parsable_pdf"])
@pytest.mark.parametrize("batch_size", [128]) @pytest.mark.parametrize("batch_size", [0, 1, 2, 64])
@pytest.mark.parametrize("input_size", [{"depth": 3, "width": 100, "height": 90}], indirect=["input_size"]) @pytest.mark.parametrize("input_size", [{"depth": 3, "width": 170, "height": 220}], indirect=["input_size"])
def test_parsable_pdf_image_extractor(image_extractor, pdf, images, metadata, 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)) 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 np.allclose(images_to_batch_tensor(images_extracted), images_to_batch_tensor(images))