code: def test_load(context_obj, sagemaker_session):
      assert context_obj.context_name
      logging.info(f"loading {context_obj.context_name}")
      loaded = context.Context.load(
          context_name=context_obj.context_name,
          sagemaker_session=sagemaker_session,
      )
      assert context_obj.context_arn == loaded.context_arn
  
file_path: sagemaker-python-sdk/tests/integ/sagemaker/lineage/test_context.py
imports: from __future__ import absolute_import
  import datetime
  import logging
  import time
  import pytest
  from sagemaker.lineage import context
  from sagemaker.lineage.query import LineageQueryDirectionEnum
label_desc: Use image node_id 5 for calling a global function with example usage: test_load(context_obj, sagemaker_session) without return types