code: def test_json_read_partitioning(ray_start_regular_shared, tmp_path):
      path = os.path.join(tmp_path, "country=us")
      os.mkdir(path)
      with open(os.path.join(path, "file1.json"), "w") as file:
          json.dump({"number": 0, "string": "foo"}, file)
      with open(os.path.join(path, "file2.json"), "w") as file:
          json.dump({"number": 1, "string": "bar"}, file)
  
      ds = ray.data.read_json(path)
  
      assert ds.take() == [
          {"number": 0, "string": "foo", "country": "us"},
          {"number": 1, "string": "bar", "country": "us"},
      ]
  
imports: import json
  import os
  import shutil
  from functools import partial
  import pandas
  import pyarrow
  import pyarrow.json
  import pytest
  from pytest_lazyfixture import lazy_fixture
  import ray
  from ray.data.block import BlockAccessor
  from ray.data.datasource import BaseFileMetadataProvider, FastFileMetadataProvider, PartitionStyle, PathPartitionFilter
  from ray.data.datasource.path_util import _unwrap_protocol
  from ray.data.tests.conftest import *
  from ray.data.tests.mock_http_server import *
  from ray.data.tests.test_partitioning import PathPartitionEncoder
  from ray.data.tests.util import Counter
  from ray.tests.conftest import *
label_desc: Use image node_id 1 for calling a global function with example usage: test_json_read_partitioning(ray_start_regular_shared, tmp_path) without return types