code: def test_numpy_read_partitioning(ray_start_regular_shared, tmp_path):
      path = os.path.join(tmp_path, "country=us", "data.npy")
      os.mkdir(os.path.dirname(path))
      np.save(path, np.arange(4).reshape([2, 2]))
  
      ds = ray.data.read_numpy(path, partitioning=Partitioning("hive"))
  
      assert ds.schema().names == ["data", "country"]
      assert [r["country"] for r in ds.take()] == ["us", "us"]
  
imports: import os
  import numpy
  import pandas
  import pyarrow
  import pytest
  from pytest_lazyfixture import lazy_fixture
  import ray
  from ray.data.datasource import BaseFileMetadataProvider, FastFileMetadataProvider, Partitioning, PartitionStyle, PathPartitionFilter
  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, extract_values
  from ray.tests.conftest import *
label_desc: Use image node_id 1 for calling a global function with example usage: test_numpy_read_partitioning(ray_start_regular_shared, tmp_path) without return types