code: def test_partition_ndarray(self):
      data = np.random.randn(10, 4)
  
      xshards = XShards.partition(data)
  
      data_parts = xshards.rdd.collect()
  
      reconstructed = np.concatenate(data_parts)
      assert np.allclose(data, reconstructed)
  
file_path: BigDL/python/orca/test/bigdl/orca/data/test_xshards_partition.py
imports: from unittest import TestCase
  import pytest
  import numpy
  from bigdl.orca.data import XShards
label_desc: Use image node_id 1 for calling the TestXshardsPartition obj's underlying member method code with example usage: obj.test_partition_ndarray() without return types