code: def test_FillDiagonalOffset(a, val, offset):
      g = extra_ops.FillDiagonalOffset()(a, val, offset)
      g_fg = FunctionGraph(outputs=[g])
  
      compare_numba_and_py(
          g_fg,
          [
              i.tag.test_value
              for i in g_fg.inputs
              if not isinstance(i, (SharedVariable, Constant))
          ],
      )
  
file_path: pytensor/tests/link/numba/test_extra_ops.py
imports: import contextlib
  import numpy
  import pytest
  import pytensor.tensor
  from pytensor import config
  from pytensor.compile.sharedvalue import SharedVariable
  from pytensor.graph.basic import Constant
  from pytensor.graph.fg import FunctionGraph
  from pytensor.tensor import extra_ops
  from tests.link.numba.test_basic import compare_numba_and_py, set_test_value
label_desc: Use image node_id 4 for calling a global function with example usage: test_FillDiagonalOffset(a, val, offset) without return types