code: def test_load_dataset_splits(
      dummy_dc_loader: load.DatasetCollectionLoader,
  ):  # pylint: disable=redefined-outer-name
      with mock.patch.object(load, "load", autospec=True) as mock_load:
          examples = tf.data.Dataset.from_tensor_slices([1, 2, 3])
          expected = {"train": examples, "test": examples}
          mock_load.return_value = [examples, examples]
          loaded_dataset = dummy_dc_loader.load_dataset(
              "c", split=["train", "test"]
          )
          mock_load.assert_called_once_with(
              name="c/e:3.5.7", with_info=False, split=["train", "test"]
          )
          assert loaded_dataset == expected
  
imports: import logging
  from unittest import mock
  import pytest
  import tensorflow
  from tensorflow_datasets import testing
  from tensorflow_datasets.core import file_adapters
  from tensorflow_datasets.core import load
  from tensorflow_datasets.core import naming
  from tensorflow_datasets.core import read_only_builder
  from tensorflow_datasets.core import registered
  from tensorflow_datasets.core import visibility
label_desc: Use image node_id 7 for calling a global function with example usage: test_load_dataset_splits(dummy_dc_loader) without return types