code: def create_context(self, context_name, context_type="Endpoint"):
      response = self.client.create_context(
          ContextName=context_name,
          Source={
              "SourceUri": "Test-context-" + context_name,
              "SourceType": "S3ETag",
              "SourceId": "Test-context-sourceId-value",
          },
          ContextType=context_type,
      )
      self.contexts.append(response["ContextArn"])
  
      return response["ContextArn"]
  
file_path: sagemaker-python-sdk/tests/integ/sagemaker/lineage/helpers.py
imports: from __future__ import absolute_import
  import uuid
  import time
  from datetime import datetime
label_desc: Use image node_id 4 for calling the LineageResourceHelper obj's underlying member method code with example usage: obj.create_context(context_name, context_type) and returns: response