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 test_publish_unpublish_model(firebase_model): assert firebase_model.published is False published_model = ml.publish_model(firebase_model.model_id) assert published_model.published is True unpublished_model = ml.unpublish_model(published_model.model_id) assert unpublished_model.published is False