from image_prediction.image_extractor.extractor import ImageExtractor, ImageMetadataPair


class ImageExtractorMock(ImageExtractor):
    def extract(self, image_container):
        for i, image in enumerate(image_container):
            yield ImageMetadataPair(image, {"image_id": i})