def test_train(import_module):
framework = Mock()
import_module.return_value = framework
_trainer.train()
import_module.assert_called_with('my_framework')
framework.entry_point.assert_called()
def test_train_no_intermediate(start_intermediate_folder_sync, import_module):
framework = Mock()
import_module.return_value = framework
_trainer.train()
import_module.assert_called_with("my_framework")
framework.entry_point.assert_called()
start_intermediate_folder_sync.asser_not_called()
def test_train_script(_exit, training_env, run):
_trainer.train()
env = training_env()
run.assert_called_with(env.module_dir, env.user_entry_point,
env.to_cmd_args(), env.to_env_vars())
_exit.assert_called_with(_trainer.SUCCESS_CODE)
def test_train_with_success(_exit, import_module):
def success():
pass
framework = Mock(entry_point=success)
import_module.return_value = framework
_trainer.train()
_exit.assert_called_with(_trainer.SUCCESS_CODE)
def test_train_fails_with_no_error_number(_exit, import_module):
def fail():
raise Exception("No errno defined.")
framework = Mock(entry_point=fail)
import_module.return_value = framework
_trainer.train()
_exit.assert_called_with(_trainer.DEFAULT_FAILURE_CODE)
def test_train_fails(_exit, import_module):
def fail():
raise OSError(os.errno.ENOENT, "No such file or directory")
framework = Mock(entry_point=fail)
import_module.return_value = framework
_trainer.train()
_exit.assert_called_with(errno.ENOENT)
def test_train_with_client_error(_exit, import_module):
def fail():
raise _errors.ClientError(os.errno.ENOENT, "No such file or directory")
framework = Mock(entry_point=fail)
import_module.return_value = framework
_trainer.train()
_exit.assert_called_with(_trainer.DEFAULT_FAILURE_CODE)
def test_train_fails_with_invalid_error_number(_exit, import_module):
class InvalidErrorNumberException(Exception):
def __init__(self, *args, **kwargs): # real signature unknown
self.errno = "invalid"
def fail():
raise InvalidErrorNumberException("No such file or directory")
framework = Mock(entry_point=fail)
import_module.return_value = framework
_trainer.train()
_exit.assert_called_with(_trainer.DEFAULT_FAILURE_CODE)
