code: def test_retrieve_feature_group_context_arns():
      with patch.object(
          SAGEMAKER_SESSION_MOCK,
          "describe_feature_group",
          return_value=FEATURE_GROUP,
      ) as fg_describe_method:
          with patch.object(
              Context,
              "load",
              side_effect=[CONTEXT_MOCK_01, CONTEXT_MOCK_02],
          ) as context_load:
              type(CONTEXT_MOCK_01).context_arn = "context-arn-fep"
              type(CONTEXT_MOCK_02).context_arn = "context-arn-fep-ver"
              result = FeatureGroupLineageEntityHandler.retrieve_feature_group_context_arns(
                  feature_group_name=FEATURE_GROUP_NAME,
                  sagemaker_session=SAGEMAKER_SESSION_MOCK,
              )
  
      assert result.name == FEATURE_GROUP_NAME
      assert result.pipeline_context_arn == "context-arn-fep"
      assert (
          result.pipeline_version_context_arn == "context-arn-fep-ver"
      )
      fg_describe_method.assert_called_once_with(
          feature_group_name=FEATURE_GROUP_NAME
      )
      context_load.assert_has_calls(
          [
              call(
                  context_name=f'{FEATURE_GROUP_NAME}-{FEATURE_GROUP["CreationTime"].strftime("%s")}'
                  f"-feature-group-pipeline",
                  sagemaker_session=SAGEMAKER_SESSION_MOCK,
              ),
              call(
                  context_name=f'{FEATURE_GROUP_NAME}-{FEATURE_GROUP["CreationTime"].strftime("%s")}'
                  f"-feature-group-pipeline-version",
                  sagemaker_session=SAGEMAKER_SESSION_MOCK,
              ),
          ]
      )
      assert 2 == context_load.call_count
  
imports: from __future__ import absolute_import
  from mock import patch, call
  from sagemaker.feature_store.feature_processor.lineage._feature_group_lineage_entity_handler import FeatureGroupLineageEntityHandler
  from sagemaker.lineage.context import Context
  from test_constants import SAGEMAKER_SESSION_MOCK, CONTEXT_MOCK_01, CONTEXT_MOCK_02, FEATURE_GROUP, FEATURE_GROUP_NAME
label_desc: Use image node_id 1 for calling a global function with example usage: test_retrieve_feature_group_context_arns() without return types