import time

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", [10])
def test_parsable_pdf_image_extractor(image_extractor, pdf, images, metadata):
    images_extracted, metadata_extracted = map(list, zip(*image_extractor(pdf)))
    # assert images_extracted == images
    assert list(metadata_extracted) == metadata