17 lines
643 B
Python
17 lines
643 B
Python
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("extractor_type", ["mock"])
|
|
@pytest.mark.parametrize("batch_size", [1, 2, 4])
|
|
def test_image_extractor_mock(image_extractor, images):
|
|
images_extracted, metadata = map(list, zip(*image_extractor(images)))
|
|
assert images_extracted == images
|
|
|
|
|
|
@pytest.mark.parametrize("extractor_type", ["parsable_pdf"])
|
|
@pytest.mark.parametrize("batch_size", [1, 2, 4])
|
|
def test_parsable_pdf_image_extractor(image_extractor, pdf, images, metadata):
|
|
images_extracted, metadata_extracted = map(list, zip(*image_extractor(images)))
|
|
assert images_extracted == images
|
|
assert metadata_extracted == metadata
|