code: def test_integer_fields(
      self,
      serialize,
      mock_endpoint,
      mock_endpoint_config,
      mock_model,
      mock_client,
  ):
      mock_endpoint.return_value = {
          "EndpointArn": "test_arn",
          "ResponseMetadata": {"HTTPStatusCode": 200},
      }
      self.sagemaker.execute(None)
      assert self.sagemaker.integer_fields == EXPECTED_INTEGER_FIELDS
      for variant in self.sagemaker.config["EndpointConfig"][
          "ProductionVariants"
      ]:
          assert variant["InitialInstanceCount"] == int(
              variant["InitialInstanceCount"]
          )
  
file_path: airflow/tests/providers/amazon/aws/operators/test_sagemaker_endpoint.py
imports: from __future__ import annotations
  from unittest import mock
  import pytest
  from botocore.exceptions import ClientError
  from airflow.exceptions import AirflowException, TaskDeferred
  from airflow.providers.amazon.aws.hooks.sagemaker import SageMakerHook
  from airflow.providers.amazon.aws.operators import sagemaker
  from airflow.providers.amazon.aws.operators.sagemaker import SageMakerEndpointOperator
  from airflow.providers.amazon.aws.triggers.sagemaker import SageMakerTrigger
label_desc: Use image node_id 2 for calling the TestSageMakerEndpointOperator obj's underlying member method code with example usage: obj.test_integer_fields(serialize, mock_endpoint, mock_endpoint_config, mock_model, mock_client) without return types