code: def test_datasetdict_from_json_splits(split, jsonl_path, tmp_path):
      if split:
          path = {split: jsonl_path}
      else:
          split = "train"
          path = {"train": jsonl_path, "test": jsonl_path}
      cache_dir = tmp_path / "cache"
      expected_features = {
          "col_1": "string",
          "col_2": "int64",
          "col_3": "float64",
      }
      dataset = JsonDatasetReader(path, cache_dir=cache_dir).read()
      _check_json_datasetdict(
          dataset, expected_features, splits=list(path.keys())
      )
      assert all(dataset[split].split == split for split in path.keys())
  
imports: import io
  import json
  import fsspec
  import pytest
  from datasets import Dataset, DatasetDict, Features, NamedSplit, Value
  from datasets.io.json import JsonDatasetReader, JsonDatasetWriter
  from ..utils import assert_arrow_memory_doesnt_increase, assert_arrow_memory_increases
label_desc: Use image node_id 11 for calling a global function with example usage: test_datasetdict_from_json_splits(split, jsonl_path, tmp_path) without return types