code: def test_jumpstart_inference_instance_type_variants():
      assert INSTANCE_TYPE_VARIANT.get_instance_specific_supported_inference_instance_types(
          "ml.p2.xlarge"
      ) == [
          "ml.p2.xlarge",
          "ml.p3.xlarge",
      ]
      assert (
          INSTANCE_TYPE_VARIANT.get_instance_specific_default_inference_instance_type(
              "ml.p2.2xlarge"
          )
          == "ml.p2.xlarge"
      )
  
      assert INSTANCE_TYPE_VARIANT.get_instance_specific_supported_inference_instance_types(
          "ml.p2.12xlarge"
      ) == [
          "ml.p2.xlarge",
          "ml.p3.xlarge",
          "ml.p5.xlarge",
      ]
      assert (
          INSTANCE_TYPE_VARIANT.get_instance_specific_default_inference_instance_type(
              "ml.p2.12xlarge"
          )
          == "ml.p5.xlarge"
      )
  
      assert (
          INSTANCE_TYPE_VARIANT.get_instance_specific_supported_inference_instance_types(
              "ml.sdfsad.12xlarge"
          )
          == []
      )
      assert (
          INSTANCE_TYPE_VARIANT.get_instance_specific_default_inference_instance_type(
              "ml.adfas.12xlarge"
          )
          is None
      )
  
      assert INSTANCE_TYPE_VARIANT.get_instance_specific_supported_inference_instance_types(
          "ml.trn1.12xlarge"
      ) == [
          "ml.inf1.2xlarge",
          "ml.inf1.xlarge",
      ]
      assert (
          INSTANCE_TYPE_VARIANT.get_instance_specific_default_inference_instance_type(
              "ml.trn1.12xlarge"
          )
          == "ml.inf1.xlarge"
      )
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/jumpstart/test_types.py
imports: from __future__ import absolute_import
  import copy
  from sagemaker.jumpstart.types import JumpStartECRSpecs, JumpStartHyperparameter, JumpStartInstanceTypeVariants, JumpStartModelSpecs, JumpStartModelHeader
  from tests.unit.sagemaker.jumpstart.constants import BASE_SPEC
label_desc: Use image node_id 6 for calling a global function with example usage: test_jumpstart_inference_instance_type_variants() without return types