code: def test_load(tmp_path):
      assert TestDatasetSource("test:" + str(tmp_path)).load() == str(
          tmp_path
      )
  
file_path: mlflow/tests/data/test_dataset_source.py
imports: import json
  import pandas
  import pytest
  import mlflow.data
  from mlflow.exceptions import MlflowException
  from tests.resources.data.dataset_source import TestDatasetSource
label_desc: Use image node_id 1 for calling a global function with example usage: test_load(tmp_path) without return types