code: def test_training_job_pipeline_execution_step(viz, sagemaker_session):
      sagemaker_session.sagemaker_client.list_trial_components.return_value = {
          "TrialComponentSummaries": [{"TrialComponentArn": "tc-arn"}]
      }
  
      get_list_associations_side_effect(sagemaker_session)
  
      step = {"Metadata": {"TrainingJob": {"Arn": "training-job-arn"}}}
  
      df = viz.show(pipeline_execution_step=step)
  
      sagemaker_session.sagemaker_client.list_trial_components.assert_called_with(
          SourceArn="training-job-arn",
      )
  
      assert_list_associations_mock_calls(sagemaker_session)
  
      pd.testing.assert_frame_equal(get_expected_dataframe(), df)
  
file_path: sagemaker-python-sdk/tests/unit/sagemaker/lineage/test_visualizer.py
imports: from __future__ import absolute_import
  import unittest.mock
  import pandas
  from collections import OrderedDict
label_desc: Use image node_id 7 for calling a global function with example usage: test_training_job_pipeline_execution_step(viz, sagemaker_session) without return types