code: def test_save(sagemaker_session):
      obj = context.Context(
          sagemaker_session,
          context_name="foo",
          description="test-description",
          properties={"k1": "v1", "k2": "v2"},
          properties_to_remove=["E"],
      )
      sagemaker_session.sagemaker_client.update_context.return_value = (
          {}
      )
      obj.save()
  
      sagemaker_session.sagemaker_client.update_context.assert_called_with(
          ContextName="foo",
          Description="test-description",
          Properties={"k1": "v1", "k2": "v2"},
          PropertiesToRemove=["E"],
      )
  
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 10 for calling a global function with example usage: test_save(sagemaker_session) without return types