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_writesToLog_ifLoggerIsInformed(self):
logger = Mock()
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.setLogger(logger)
target.load()
logger.log.assert_has_calls([call('Reading file %s' % name) for name in ['file.jpg', 'file2.jpg', 'file3.jpg']])
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_writesToLog_ifLoggerIsInformed(self):
logger = Mock()
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.setLogger(logger)
target.load()
logger.log.assert_has_calls([
call('Reading file %s' % name)
for name in ['file.jpg', 'file2.jpg', 'file3.jpg']
])
