code: def test_register_dataset_source_and_resolve(tmp_path):
      registry = DatasetSourceRegistry()
      registry.register(TestDatasetSource)
  
      uri = "test:" + str(tmp_path)
      resolved_source = registry.resolve(uri)
      assert isinstance(resolved_source, TestDatasetSource)
      # Verify that the TestDatasetSource is constructed with the correct URI
      assert resolved_source.uri == uri
  
file_path: mlflow/tests/data/test_dataset_source_registry.py
imports: from typing import Any
  from unittest import mock
  import pytest
  from mlflow.data.dataset_source_registry import DatasetSourceRegistry
  from mlflow.exceptions import MlflowException
  from tests.resources.data.dataset_source import TestDatasetSource
label_desc: Use image node_id 2 for calling a global function with example usage: test_register_dataset_source_and_resolve(tmp_path) without return types