code: def test_image(sagemaker_session):
      ntm = NTM(sagemaker_session=sagemaker_session, **ALL_REQ_ARGS)
      assert (
          image_uris.retrieve("ntm", REGION) == ntm.training_image_uri()
      )
  
imports: from __future__ import absolute_import
  import pytest
  from mock import Mock, patch
  from sagemaker import image_uris
  from sagemaker.amazon.ntm import NTM, NTMPredictor
  from sagemaker.amazon.amazon_estimator import RecordSet
  from sagemaker.session_settings import SessionSettings
label_desc: Use image node_id 5 for calling a global function with example usage: test_image(sagemaker_session) without return types