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