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)