def CreateWithImageReader(imageReader, sourceFolder, log = False, loadOnlyClasses = None, getLabelFunction = None):
if getLabelFunction is None:
getLabelFunction = GetLabelFromFirstChars(nChars=2)
imageDataSource = ImageDataSource(fileSystem = FileSystem(),
imageReader = imageReader,
sourceFolder = sourceFolder)
if log:
imageDataSource.setLogger(TextLogger())
if loadOnlyClasses is not None:
imageDataSource.setFilenamePredicate(LabelInFilenamePredicate(getLabelFunction, loadOnlyClasses))
return imageDataSource
def test_appliesFilter_ifFilterInformed(self):
predicate = lambda x: x in ['file.jpg', 'file3.jpg']
self.fileSystem.listDir.return_value = ['file.jpg', 'file2.jpg', 'file3.jpg']
self.fileSystem.isFile.return_value = True
self.imageReader.read.return_value = Mock()
target = ImageDataSource(fileSystem = self.fileSystem,
imageReader = self.imageReader,
sourceFolder = self.sourceFolder)
target.setFilenamePredicate(predicate)
target.load()
self.imageReader.read.assert_has_calls([call(self.sourceFolder,'file.jpg'), call(self.sourceFolder,'file3.jpg')])
self.assertEqual(len(self.imageReader.read.call_args_list), 2)
def CreateWithImageReader(imageReader,
sourceFolder,
log=False,
loadOnlyClasses=None,
getLabelFunction=None):
if getLabelFunction is None:
getLabelFunction = GetLabelFromFirstChars(nChars=2)
imageDataSource = ImageDataSource(fileSystem=FileSystem(),
imageReader=imageReader,
sourceFolder=sourceFolder)
if log:
imageDataSource.setLogger(TextLogger())
if loadOnlyClasses is not None:
imageDataSource.setFilenamePredicate(
LabelInFilenamePredicate(getLabelFunction, loadOnlyClasses))
return imageDataSource
def test_appliesFilter_ifFilterInformed(self):
predicate = lambda x: x in ['file.jpg', 'file3.jpg']
self.fileSystem.listDir.return_value = [
'file.jpg', 'file2.jpg', 'file3.jpg'
]
self.fileSystem.isFile.return_value = True
self.imageReader.read.return_value = Mock()
target = ImageDataSource(fileSystem=self.fileSystem,
imageReader=self.imageReader,
sourceFolder=self.sourceFolder)
target.setFilenamePredicate(predicate)
target.load()
self.imageReader.read.assert_has_calls([
call(self.sourceFolder, 'file.jpg'),
call(self.sourceFolder, 'file3.jpg')
])
self.assertEqual(len(self.imageReader.read.call_args_list), 2)
