def METHOD_NAME(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)