def test_main_prepare_flickr_limit(mocker):
"""It should run the Flickr preparer with specified limit."""
flickr_mock = mocker.patch('mila.prepare.flickr.run')
cli.main(['prepare', 'flickr', '--user', 'abcd', '--tags', 'cat', '--limit', '123'])
flickr_mock.assert_called_once()
flickr_mock.assert_called_with('abcd', 'cat', 123)
def test_main_explore(mocker):
"""It should launch quiver"""
mocker.patch('mila.predict.load', return_value=('FAKE_MODEL', {'classes': ['cat', 'dog']}))
mocker.patch('tempfile.mkdtemp', return_value='FAKE_TMP_DIR')
quiver_mock = mocker.patch('quiver_engine.server.launch')
cli.main(['explore', 'images/all/cat', 'output/mymodeldir'])
quiver_mock.assert_called_with(
'FAKE_MODEL',
classes=['cat', 'dog'],
input_folder='images/all/cat',
temp_folder='FAKE_TMP_DIR',
std=[255, 255, 255])
def test_main_train_simple_with_params(mocker):
"""It should run the training with given parameters."""
simple_mock = mocker.patch('mila.train.simple.train')
cli.main([
'train',
'simple',
'--epochs',
'123',
'--outputdir',
'./anotherplace/mymodeldir',
'--imagesize',
'111,222',
'--batchsize',
'456',
'--classweights',
'--imagevariations',
'--debug'])
simple_mock.assert_called_with((111, 222), 123, 456, './anotherplace/mymodeldir', debug=True, use_class_weights=True, use_image_variations=True)
def test_main_predict_defaults(mocker):
"""It should predict with default parameters."""
simple_mock = mocker.patch('mila.predict.predict')
cli.main(['predict', 'images/cat.jpg', 'output/mymodeldir'])
simple_mock.assert_called_with('images/cat.jpg', 'output/mymodeldir')
def test_main_train_mobilenet_defaults(mocker):
"""It should run the mobilenet training with default parameters."""
simple_mock = mocker.patch('mila.train.mobilenet.train')
cli.main(['train', 'mobilenet'])
simple_mock.assert_called_with(10, 32, './output/mobilenet', use_class_weights=False, use_image_variations=False)
def test_main_train_simple_defaults(mocker):
"""It should run the training with default parameters."""
simple_mock = mocker.patch('mila.train.simple.train')
cli.main(['train', 'simple'])
simple_mock.assert_called_with((32, 32), 10, 32, './output/simple', debug=False, use_class_weights=False, use_image_variations=False)
def test_main_prepare_test_split(mocker):
"""It should run the Test split run function."""
split_mock = mocker.patch('mila.prepare.train_split.run')
cli.main(['prepare', 'traindata', '--equalsplits'])
split_mock.assert_called_with(equal_splits=True)