code: def test_set_tags(sagemaker_session):
      sagemaker_session.sagemaker_client.create_context.return_value = {
          "ContextArn": "context-arn",
      }
      obj = context.Context.create(
          context_name="foo",
          source_uri="bar",
          sagemaker_session=sagemaker_session,
      )
      tags = [
          {"Key": "foo1", "Value": "bar1"},
          {"Key": "foo2", "Value": "bar2"},
      ]
      sagemaker_session.sagemaker_client.add_tags.return_value = {
          "Tags": tags
      }
      obj.set_tags(tags=tags)
  
      sagemaker_session.sagemaker_client.add_tags.assert_called_with = {
          "ResourceArn": "context-arn",
          "Tags": [
              {"Key": "foo1", "Value": "bar1"},
              {"Key": "foo2", "Value": "bar2"},
          ],
      }
  
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 5 for calling a global function with example usage: test_set_tags(sagemaker_session) without return types