def test_delete_model(firebase_model): ml.delete_model(firebase_model.model_id) # Second delete of same model will fail with pytest.raises(exceptions.NotFoundError) as excinfo: ml.delete_model(firebase_model.model_id) check_firebase_error(excinfo, 404, 'Requested entity was not found.')
def test_delete_model(self): recorder = instrument_ml_service(status=200, payload=EMPTY_RESPONSE) ml.delete_model(MODEL_ID_1) # no response for delete assert len(recorder) == 1 assert recorder[0].method == 'DELETE' assert recorder[0].url == TestDeleteModel._url(PROJECT_ID, MODEL_ID_1) assert recorder[0].headers[HEADER_CLIENT_KEY] == HEADER_CLIENT_VALUE
def test_get_non_existing_model(firebase_model): # Get a valid model_id that no longer exists ml.delete_model(firebase_model.model_id) with pytest.raises(exceptions.NotFoundError) as excinfo: ml.get_model(firebase_model.model_id) check_firebase_error(excinfo, 404, 'Requested entity was not found.')
def _clean_up_model(model): try: # Try to delete the model. # Some tests delete the model as part of the test. ml.delete_model(model.model_id) except exceptions.NotFoundError: pass
def test_update_non_existing_model(firebase_model): ml.delete_model(firebase_model.model_id) firebase_model.tags = ['tag987'] with pytest.raises(exceptions.NotFoundError) as excinfo: ml.update_model(firebase_model) check_operation_error( excinfo, 'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name')))
def test_delete_model_error(self): recorder = instrument_ml_service(status=404, payload=ERROR_RESPONSE_NOT_FOUND) with pytest.raises(exceptions.NotFoundError) as excinfo: ml.delete_model(MODEL_ID_1) check_firebase_error(excinfo, ERROR_STATUS_NOT_FOUND, ERROR_CODE_NOT_FOUND, ERROR_MSG_NOT_FOUND) assert len(recorder) == 1 assert recorder[0].method == 'DELETE' assert recorder[0].url == self._url(PROJECT_ID, MODEL_ID_1) assert recorder[0].headers[HEADER_CLIENT_KEY] == HEADER_CLIENT_VALUE
def test_publish_unpublish_non_existing_model(firebase_model): ml.delete_model(firebase_model.model_id) with pytest.raises(exceptions.NotFoundError) as excinfo: ml.publish_model(firebase_model.model_id) check_operation_error( excinfo, 'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name'))) with pytest.raises(exceptions.NotFoundError) as excinfo: ml.unpublish_model(firebase_model.model_id) check_operation_error( excinfo, 'Model \'{0}\' was not found'.format(firebase_model.as_dict().get('name')))
def delete_model(model_id): """Delete a model from the project.""" ml.delete_model(model_id)
def evaluate(): app = firebase_admin.initialize_app(testutils.MockCredential(), name='no_project_id') with pytest.raises(ValueError): ml.delete_model(MODEL_ID_1, app)
def test_delete_model_validation_errors(self, model_id, exc_type): with pytest.raises(exc_type) as excinfo: ml.delete_model(model_id) check_error(excinfo, exc_type)