code: def test_load(sagemaker_session):
      now = datetime.datetime.now(datetime.timezone.utc)
  
      sagemaker_session.sagemaker_client.describe_context.return_value = {
          "ContextName": "A",
          "Source": {
              "SourceUri": "B",
              "SourceType": "test-source-type",
          },
          "ContextArn": "D",
          "ContextType": "E",
          "Properties": {"F": "FValue", "G": "GValue"},
          "CreationTime": now,
          "CreatedBy": {},
          "LastModifiedTime": now,
          "LastModifiedBy": {},
      }
      obj = context.Context.load(
          context_name="A", sagemaker_session=sagemaker_session
      )
      sagemaker_session.sagemaker_client.describe_context.assert_called_with(
          ContextName="A"
      )
      assert "A" == obj.context_name
      assert "B" == obj.source.source_uri
      assert "test-source-type" == obj.source.source_type
      assert "D" == obj.context_arn
      assert "E" == obj.context_type
      assert {"F": "FValue", "G": "GValue"} == obj.properties
  
imports: from __future__ import absolute_import
  import datetime
  import unittest.mock
  import pytest
  from sagemaker.lineage import context, _api_types
  from sagemaker.lineage.action import Action
  from sagemaker.lineage.lineage_trial_component import LineageTrialComponent
  from sagemaker.lineage.query import LineageQueryDirectionEnum
label_desc: Use image node_id 6 for calling a global function with example usage: test_load(sagemaker_session) without return types