def test_id_from_image(self): source_dataset = Dataset.from_iterable([ DatasetItem(id=1, image='path.jpg'), DatasetItem(id=2), ]) target_dataset = Dataset.from_iterable([ DatasetItem(id='path', image='path.jpg'), DatasetItem(id=2), ]) actual = transforms.IdFromImageName(source_dataset) compare_datasets(self, target_dataset, actual)
def test_id_from_image(self): class SrcExtractor(Extractor): def __iter__(self): return iter([ DatasetItem(id=1, image='path.jpg'), DatasetItem(id=2), ]) class DstExtractor(Extractor): def __iter__(self): return iter([ DatasetItem(id='path', image='path.jpg'), DatasetItem(id=2), ]) actual = transforms.IdFromImageName(SrcExtractor()) compare_datasets(self, DstExtractor(), actual)