code: def test_create(sagemaker_session):
      sagemaker_session.sagemaker_client.create_context.return_value = {
          "ContextArn": "bazz",
      }
      obj = context.Context.create(
          context_name="foo",
          source_uri="bar",
          source_type="test-source-type",
          context_type="test-context-type",
          description="test-description",
          properties={"k1": "v1", "k2": "v2"},
          sagemaker_session=sagemaker_session,
      )
      sagemaker_session.sagemaker_client.create_context.assert_called_with(
          ContextName="foo",
          Source={"SourceUri": "bar", "SourceType": "test-source-type"},
          ContextType="test-context-type",
          Description="test-description",
          Properties={"k1": "v1", "k2": "v2"},
      )
      assert "foo" == obj.context_name
      assert "bar" == obj.source.source_uri
      assert "test-source-type" == obj.source.source_type
      assert "test-context-type" == obj.context_type
      assert "test-description" == obj.description
      assert {"k1": "v1", "k2": "v2"} == obj.properties
      assert "bazz" == obj.context_arn
  
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 2 for calling a global function with example usage: test_create(sagemaker_session) without return types