code: def sagemaker_session():
      boto_mock = MagicMock(name="boto_session")
      client_mock = MagicMock()
      client_mock.get_caller_identity.return_value = {
          "UserId": "mock_user_id",
          "Account": "012345678910",
          "Arn": "arn:aws:iam::012345678910:user/mock-user",
      }
      boto_mock.client.return_value = client_mock
      ims = sagemaker.Session(boto_session=boto_mock)
      ims.default_bucket = Mock(
          name="default_bucket", return_value=BUCKET_NAME
      )
      return ims
  
imports: from __future__ import absolute_import
  import os
  from mock import MagicMock, Mock
  import pytest
  import sagemaker
label_desc: Use image node_id 1 for calling a global function with example usage: sagemaker_session() and returns: ims