def test_update_model(self, mocker):
mock_method = mocker.patch.object(MlflowClient,
"rename_registered_model")
client = ModelsClient()
name = "simple-nn-model"
new_name = "new-simple-nn-model"
client.update_model(name, new_name)
mock_method.assert_called_once_with(name="simple-nn-model",
new_name="new-simple-nn-model")
def test_update_model(self):
client = ModelsClient()
name = "simple-nn-model"
new_name = "new-simple-nn-model"
client.update_model(name, new_name)