code: def test_model_image(sagemaker_session):
      kmeans = KMeans(
          sagemaker_session=sagemaker_session, **ALL_REQ_ARGS
      )
      data = RecordSet(
          "s3://{}/{}".format(BUCKET_NAME, PREFIX),
          num_records=1,
          feature_dim=FEATURE_DIM,
          channel="train",
      )
      kmeans.fit(data, MINI_BATCH_SIZE)
  
      model = kmeans.create_model()
      assert image_uris.retrieve("kmeans", REGION) == model.image_uri
  
imports: from __future__ import absolute_import
  import pytest
  from mock import Mock, patch
  from sagemaker import image_uris
  from sagemaker.amazon.kmeans import KMeans, KMeansPredictor
  from sagemaker.amazon.amazon_estimator import RecordSet
  from sagemaker.session_settings import SessionSettings
label_desc: Use image node_id 15 for calling a global function with example usage: test_model_image(sagemaker_session) without return types