code: def test_trained_models(sagemaker_session):
      # dataset artifact ---- tc --- artifact model
      dataset_artifact_obj = artifact.DatasetArtifact(
          sagemaker_session,
          artifact_arn="dataset-artifact-arn",
          artifact_name="dataset-artifact-name",
      )
      sagemaker_session.sagemaker_client.list_associations.side_effect = [
          {
              "AssociationSummaries": [
                  {
                      "SourceArn": dataset_artifact_obj.artifact_arn,
                      "SourceName": "X1",
                      "DestinationArn": "experiment-trial-component",
                      "DestinationName": "Y1",
                      "SourceType": "C1",
                      "DestinationType": "ModelDeployment",
                      "AssociationType": "E1",
                      "CreationTime": None,
                      "CreatedBy": None,
                  }
              ],
          },
          {
              "AssociationSummaries": [
                  {
                      "SourceArn": "experiment-trial-component",
                      "SourceName": "X2",
                      "DestinationArn": "B2",
                      "DestinationName": "Y2",
                      "SourceType": "C2",
                      "DestinationType": "Context",
                      "AssociationType": "E2",
                      "CreationTime": None,
                      "CreatedBy": None,
                  }
              ]
          },
      ]
  
      model_list = dataset_artifact_obj.trained_models()
      expected_calls = [
          unittest.mock.call(
              SourceArn=dataset_artifact_obj.artifact_arn
          ),
          unittest.mock.call(
              SourceArn="experiment-trial-component",
              DestinationType="Context",
          ),
      ]
      assert (
          expected_calls
          == sagemaker_session.sagemaker_client.list_associations.mock_calls
      )
      expected_model_list = [
          _api_types.AssociationSummary(
              source_arn="experiment-trial-component",
              source_name="X2",
              destination_arn="B2",
              destination_name="Y2",
              source_type="C2",
              destination_type="Context",
              association_type="E2",
              creation_time=None,
              created_by=None,
          )
      ]
      assert expected_model_list == model_list
  
imports: from __future__ import absolute_import
  import unittest.mock
  import pytest
  from sagemaker.lineage import artifact, _api_types
label_desc: Use image node_id 2 for calling a global function with example usage: test_trained_models(sagemaker_session) without return types