code: def test_training_image_uri(
      sagemaker_session, xgboost_framework_version
  ):
      xgboost = XGBoost(
          entry_point=SCRIPT_PATH,
          role=ROLE,
          framework_version=xgboost_framework_version,
          sagemaker_session=sagemaker_session,
          instance_type=INSTANCE_TYPE,
          instance_count=1,
          py_version=PYTHON_VERSION,
      )
  
      assert (
          _get_full_image_uri(xgboost_framework_version)
          in xgboost.training_image_uri()
      )
  
imports: from __future__ import absolute_import
  import logging
  import json
  import os
  import pytest
  from mock import Mock
  from mock import patch
  from packaging.version import Version
  from sagemaker.fw_utils import UploadedCode
  from sagemaker.session_settings import SessionSettings
  from sagemaker.xgboost import XGBoost, XGBoostModel, XGBoostPredictor
label_desc: Use image node_id 15 for calling a global function with example usage: test_training_image_uri(sagemaker_session, xgboost_framework_version) without return types