code: def test_blockwise_new_axes_2():
      x = da.ones((2, 2), chunks=(1, 1))
  
      def func(x):
          return np.stack([x, -x], axis=-1)
  
      y = da.blockwise(
          func,
          ("x", "y", "sign"),
          x,
          ("x", "y"),
          dtype=x.dtype,
          concatenate=True,
          new_axes={"sign": 2},
      )
  
      assert_eq(y, y)
  
imports: from __future__ import annotations
  import collections
  from operator import add
  import numpy
  import pytest
  import dask
  import dask.array
  from dask.array.utils import assert_eq
  from dask.blockwise import _BLOCKWISE_DEFAULT_PREFIX, Blockwise, _unique_dep, index_subs, optimize_blockwise, rewrite_blockwise
  from dask.highlevelgraph import HighLevelGraph
  from dask.utils_test import dec, hlg_layer_topological, inc
label_desc: Use image node_id 14 for calling a global function with example usage: test_blockwise_new_axes_2() and returns: np