code: def test_distributed_array_assemble(ray_start_2_cpus, reload_modules):
      a = ra.ones.remote([da.BLOCK_SIZE, da.BLOCK_SIZE])
      b = ra.zeros.remote([da.BLOCK_SIZE, da.BLOCK_SIZE])
      x = da.DistArray(
          [2 * da.BLOCK_SIZE, da.BLOCK_SIZE], np.array([[a], [b]])
      )
      assert_equal(
          x.assemble(),
          np.vstack(
              [
                  np.ones([da.BLOCK_SIZE, da.BLOCK_SIZE]),
                  np.zeros([da.BLOCK_SIZE, da.BLOCK_SIZE]),
              ]
          ),
      )
  
file_path: ray/python/ray/tests/test_array.py
imports: from importlib import reload
  import numpy
  from numpy.testing import assert_equal, assert_almost_equal
  import pytest
  import sys
  import ray
  import ray.experimental.array.remote
  import ray.experimental.array.distributed
  import ray.cluster_utils
label_desc: Use image node_id 3 for calling a global function with example usage: test_distributed_array_assemble(ray_start_2_cpus, reload_modules) without return types