code: def test_update_endpoint_all_args():
      predictor = empty_predictor()
      predictor_async = AsyncPredictor(predictor=predictor)
  
      predictor_async.update_endpoint()
  
      new_instance_count = 2
      new_instance_type = "ml.c4.xlarge"
      new_accelerator_type = "ml.eia1.medium"
      new_model_name = "new-model"
      new_tags = [{"Key": "foo", "Value": "bar"}]
      new_kms_key = "new-key"
      new_data_capture_config_dict = {}
  
      predictor_async.update_endpoint(
          initial_instance_count=new_instance_count,
          instance_type=new_instance_type,
          accelerator_type=new_accelerator_type,
          model_name=new_model_name,
          tags=new_tags,
          kms_key=new_kms_key,
          data_capture_config_dict=new_data_capture_config_dict,
          wait=False,
      )
  
      predictor.update_endpoint.assert_called_with(
          initial_instance_count=new_instance_count,
          instance_type=new_instance_type,
          accelerator_type=new_accelerator_type,
          model_name=new_model_name,
          tags=new_tags,
          kms_key=new_kms_key,
          data_capture_config_dict=new_data_capture_config_dict,
          wait=False,
      )
  
imports: from __future__ import absolute_import
  import pytest
  from mock import Mock
  from sagemaker.async_inference.waiter_config import WaiterConfig
  from sagemaker.predictor import Predictor
  from sagemaker.predictor_async import AsyncPredictor
  from sagemaker.exceptions import AsyncInferenceModelError, PollingTimeoutError
label_desc: Use image node_id 15 for calling a global function with example usage: test_update_endpoint_all_args() without return types