code: def test_scalar_arrays(ray_start_regular_shared):
      data = np.array([1, 2, 3])
      ds = ray.data.range(2, parallelism=1)
      ds = ds.map(lambda x: {"output": data})
      output = ds.take_batch()["output"]
      assert_structure_equals(
          output, np.array([[1, 2, 3], [1, 2, 3]], dtype=np.int64)
      )
  
imports: import numpy
  import pytest
  import torch
  import ray
  from ray.air.util.tensor_extensions.utils import create_ragged_ndarray
  from ray.data.tests.conftest import *
  from ray.tests.conftest import *
label_desc: Use image node_id 13 for calling a global function with example usage: test_scalar_arrays(ray_start_regular_shared) without return types