def test_delete_model(self):
num_models = len(mr.list_models())
model = mr.get_model(self.model_name)
mr.delete_model(model)
model = mr.get_model(model.id)
assert model is None
all_models = mr.list_models()
assert len(all_models) == num_models - 1
def test_create_model(self):
# Count the current # of models in the environment
models = mr.list_models()
assert isinstance(models, list)
model_count = len(models)
# Create a new model
model = mr.create_model(self.model_name, self.project_name, force=True)
assert isinstance(model, RestObj)
assert model.name == self.model_name
# Total number of models should have increased
models = mr.list_models()
assert len(models) == model_count + 1
def test_delete_model(self):
model_count = len(mr.list_models())
model1 = mr.get_model(self.model_name)
assert isinstance(model1, RestObj)
model2 = mr.get_model(model1.id)
assert model1.id == model2.id
mr.delete_model(model1)
model3 = mr.get_model(model1.id)
assert model3 is None
all_models = mr.list_models()
assert len(all_models) == model_count - 1