Python _delete_registered_modelの例

コード例 #1

ファイル: test_handlers.py プロジェクト: xinluh/mlflow

def test_delete_registered_model(mock_get_request_message,
                                 mock_model_registry_store):
    name = "model_1"
    mock_get_request_message.return_value = DeleteRegisteredModel(name=name)
    _delete_registered_model()
    _, args = mock_model_registry_store.delete_registered_model.call_args
    assert args == {"name": name}

コード例 #2

def test_delete_registered_model(mock_get_request_message,
                                 mock_model_registry_store):
    rm = RegisteredModel("model_1")
    mock_get_request_message.return_value = DeleteRegisteredModel(
        registered_model=rm.to_proto())
    _delete_registered_model()
    args, _ = mock_model_registry_store.delete_registered_model.call_args
    assert args == (rm, )