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