code: def test_load_makes_expected_mlflow_artifacts_download_call(
      source_uri, source_type, tmp_path
  ):
      dataset_source = resolve_dataset_source(source_uri)
      assert dataset_source._get_source_type() == source_type
  
      with mock.patch(
          "mlflow.data.artifact_dataset_sources.download_artifacts"
      ) as download_imp_mock:
          dataset_source.load()
          download_imp_mock.assert_called_once_with(
              artifact_uri=source_uri, dst_path=None
          )
  
      with mock.patch(
          "mlflow.data.artifact_dataset_sources.download_artifacts"
      ) as download_imp_mock:
          dataset_source.load(dst_path=str(tmp_path))
          download_imp_mock.assert_called_once_with(
              artifact_uri=source_uri, dst_path=str(tmp_path)
          )
  
file_path: mlflow/tests/data/test_artifact_dataset_sources.py
imports: import json
  import os
  from unittest import mock
  import pytest
  from mlflow.data.dataset_source_registry import get_dataset_source_from_json, resolve_dataset_source
  from mlflow.data.filesystem_dataset_source import FileSystemDatasetSource
  from mlflow.store.artifact.s3_artifact_repo import S3ArtifactRepository
label_desc: Use image node_id 3 for calling a global function with example usage: test_load_makes_expected_mlflow_artifacts_download_call(source_uri, source_type, tmp_path) without return types