code: def test_jax_Alloc():
      x = ptb.alloc(0.0, 2, 3)
      x_fg = FunctionGraph([], [x])
  
      _, [jax_res] = compare_jax_and_py(x_fg, [])
  
      assert jax_res.shape == (2, 3)
  
      x = ptb.alloc(1.1, 2, 3)
      x_fg = FunctionGraph([], [x])
  
      compare_jax_and_py(x_fg, [])
  
      x = ptb.AllocEmpty("float32")(2, 3)
      x_fg = FunctionGraph([], [x])
  
      def compare_shape_dtype(x, y):
          (x,) = x
          (y,) = y
          return x.shape == y.shape and x.dtype == y.dtype
  
      compare_jax_and_py(x_fg, [], assert_fn=compare_shape_dtype)
  
      a = scalar("a")
      x = ptb.alloc(a, 20)
      x_fg = FunctionGraph([a], [x])
  
      compare_jax_and_py(x_fg, [10.0])
  
      a = vector("a")
      x = ptb.alloc(a, 20, 10)
      x_fg = FunctionGraph([a], [x])
  
      compare_jax_and_py(x_fg, [np.ones(10, dtype=config.floatX)])
  
imports: import numpy
  import pytest
  from pytensor.compile import get_mode
  import jax.errors
  import pytensor
  import pytensor.tensor.basic
  from pytensor.configdefaults import config
  from pytensor.graph.fg import FunctionGraph
  from pytensor.graph.op import get_test_value
  from pytensor.tensor.type import iscalar, matrix, scalar, vector
  from tests.link.jax.test_basic import compare_jax_and_py
  from tests.tensor.test_basic import TestAlloc
label_desc: Use image node_id 1 for calling a global function with example usage: test_jax_Alloc() and returns: unknown