code: def test_save(context_obj, sagemaker_session):
      context_obj.description = "updated description"
      context_obj.properties = {"k3": "v3"}
      context_obj.properties_to_remove = ["k1"]
  
      context_obj.save()
  
      loaded = context.Context.load(
          context_name=context_obj.context_name,
          sagemaker_session=sagemaker_session,
      )
  
      assert "updated description" == loaded.description
      assert {"k3": "v3"} == loaded.properties
  
file_path: sagemaker-python-sdk/tests/integ/sagemaker/lineage/test_context.py
imports: from __future__ import absolute_import
  import datetime
  import logging
  import time
  import pytest
  from sagemaker.lineage import context
  from sagemaker.lineage.query import LineageQueryDirectionEnum
label_desc: Use image node_id 4 for calling a global function with example usage: test_save(context_obj, sagemaker_session) without return types