code: def test_polygamma():
      n = vector("n", dtype="int32")
      x = vector("x", dtype="float32")
      out = polygamma(n, x)
      fg = FunctionGraph([n, x], [out])
      compare_jax_and_py(
          fg,
          [
              np.array([0, 1, 2]).astype("int32"),
              np.array([0.5, 0.9, 2.5]).astype("float32"),
          ],
      )
  
imports: import numpy
  import pytest
  import pytensor.scalar.basic
  import pytensor.tensor
  from pytensor.configdefaults import config
  from pytensor.graph.fg import FunctionGraph
  from pytensor.graph.op import get_test_value
  from pytensor.scalar.basic import Composite
  from pytensor.tensor import as_tensor
  from pytensor.tensor.elemwise import Elemwise
  from pytensor.tensor.math import all
  from pytensor.tensor.math import betaincinv, cosh, erf, erfc, erfcinv, erfcx, erfinv, gammainccinv, gammaincinv, iv, log, log1mexp, polygamma, psi, sigmoid, softplus, tri_gamma
  from pytensor.tensor.type import matrix, scalar, vector
  from tests.link.jax.test_basic import compare_jax_and_py
  from pytensor.link.jax.dispatch import jax_funcify
label_desc: Use image node_id 15 for calling a global function with example usage: test_polygamma() without return types