code: def test_jumpstart_sdk_environment_variables(patched_get_model_specs):
      patched_get_model_specs.side_effect = get_spec_from_base_spec
  
      model_id = "pytorch-eqa-bert-base-cased"
      region = "us-west-2"
  
      vars = environment_variables.retrieve_default(
          region=region,
          model_id=model_id,
          model_version="*",
          include_aws_sdk_env_vars=False,
          sagemaker_session=mock_session,
      )
      assert vars == {
          "ENDPOINT_SERVER_TIMEOUT": "3600",
          "MODEL_CACHE_ROOT": "/opt/ml/model",
          "SAGEMAKER_ENV": "1",
          "SAGEMAKER_MODEL_SERVER_WORKERS": "1",
          "SAGEMAKER_PROGRAM": "inference.py",
      }
  
      patched_get_model_specs.assert_called_once_with(
          region=region,
          model_id=model_id,
          version="*",
          s3_client=mock_client,
      )
  
      patched_get_model_specs.reset_mock()
  
      vars = environment_variables.retrieve_default(
          region=region,
          model_id=model_id,
          model_version="1.*",
          include_aws_sdk_env_vars=False,
          sagemaker_session=mock_session,
      )
      assert vars == {
          "ENDPOINT_SERVER_TIMEOUT": "3600",
          "MODEL_CACHE_ROOT": "/opt/ml/model",
          "SAGEMAKER_ENV": "1",
          "SAGEMAKER_MODEL_SERVER_WORKERS": "1",
          "SAGEMAKER_PROGRAM": "inference.py",
      }
  
      patched_get_model_specs.assert_called_once_with(
          region=region,
          model_id=model_id,
          version="1.*",
          s3_client=mock_client,
      )
  
      patched_get_model_specs.reset_mock()
  
      with pytest.raises(KeyError):
          environment_variables.retrieve_default(
              region=region,
              model_id="blah",
              model_version="*",
              include_aws_sdk_env_vars=False,
          )
  
      with pytest.raises(ValueError):
          environment_variables.retrieve_default(
              region="mars-south-1",
              model_id=model_id,
              model_version="*",
              include_aws_sdk_env_vars=False,
          )
  
      with pytest.raises(ValueError):
          environment_variables.retrieve_default(
              model_version="*",
              include_aws_sdk_env_vars=False,
          )
  
      with pytest.raises(ValueError):
          environment_variables.retrieve_default(
              model_id=model_id,
              include_aws_sdk_env_vars=False,
          )
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/environment_variables/jumpstart/test_default.py
imports: from __future__ import absolute_import
  import boto3
  from mock.mock import patch, Mock
  import pytest
  from sagemaker import environment_variables
  from tests.unit.sagemaker.jumpstart.utils import get_spec_from_base_spec, get_special_model_spec
label_desc: Use image node_id 2 for calling a global function with example usage: test_jumpstart_sdk_environment_variables(patched_get_model_specs) without return types