code: def test_create_with_tags(sagemaker_session):
      sagemaker_session.sagemaker_client.create_context.return_value = {
          "ContextArn": "bazz",
      }
  
      tags = [{"Key": "foo", "Value": "bar"}]
      context.Context.create(
          context_name="foo",
          source_uri="bar",
          tags=tags,
          sagemaker_session=sagemaker_session,
      )
  
      sagemaker_session.sagemaker_client.create_context.assert_called_with(
          ContextName="foo",
          Source={"SourceUri": "bar"},
          Tags=[{"Key": "foo", "Value": "bar"}],
      )
  
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 3 for calling a global function with example usage: test_create_with_tags(sagemaker_session) without return types