def test_dynamic_load_model_with_model_is_none():
LATEST_MODEL_NAME = 'latest_model_name'
def mocked_init(*args, **kwargs):
return None
def mocked_search_for_models(self):
pass
def mocked_latest_project_model(self):
return LATEST_MODEL_NAME
with mock.patch.object(Project, "__init__", mocked_init):
with mock.patch.object(Project, "_search_for_models",
mocked_search_for_models):
with mock.patch.object(Project, "_latest_project_model",
mocked_latest_project_model):
project = Project()
project._models = ()
project.pull_models = None
result = project._dynamic_load_model(None)
assert result == LATEST_MODEL_NAME
def test_dynamic_load_model_with_exists_model():
MODEL_NAME = 'model_name'
def mocked_init(*args, **kwargs):
return None
with mock.patch.object(Project, "__init__", mocked_init):
project = Project()
project._models = (MODEL_NAME, )
project.pull_models = None
result = project._dynamic_load_model(MODEL_NAME)
assert result == MODEL_NAME
def test_dynamic_load_model_with_refresh_exists_model():
MODEL_NAME = 'model_name'
def mocked_init(*args, **kwargs):
return None
def mocked_search_for_models(self):
self._models = (MODEL_NAME, )
with mock.patch.object(Project, "__init__", mocked_init):
with mock.patch.object(Project, '_search_for_models',
mocked_search_for_models):
project = Project()
project._models = ()
project.pull_models = None
result = project._dynamic_load_model(MODEL_NAME)
assert result == MODEL_NAME