import pytest


@pytest.mark.parametrize("extractor_type", ["mock"])
@pytest.mark.parametrize("batch_size", [1, 2, 4])
def test_image_extraction(image_extractor, images):
    images_extracted, metadata = map(list, zip(*image_extractor(images)))
    assert images_extracted == images