code: def test_fallback_log_jac_det_undefined_ndim(self, ndim):
      class SquareTransform(Transform):
          name = "square"
          ndim_supp = ndim
  
          def forward(self, value, *inputs):
              return pt.power(value, 2)
  
          def backward(self, value, *inputs):
              return pt.sqrt(value)
  
      with pytest.raises(
          NotImplementedError,
          match=r"only implemented for ndim_supp in \(0, 1\)",
      ):
          SquareTransform().log_jac_det(0)
  
imports: import numpy
  import pytensor
  import pytensor.tensor
  import pytest
  import scipy
  import scipy.special
  from pytensor.graph.basic import equal_computations
  from pymc.distributions.continuous import Cauchy, ChiSquared
  from pymc.logprob.basic import conditional_logp, icdf, logcdf, logp
  from pymc.logprob.transforms import ArccoshTransform, ArcsinhTransform, ArctanhTransform, ChainedTransform, CoshTransform, ErfcTransform, ErfcxTransform, ErfTransform, ExpTransform, LocTransform, LogTransform, ScaleTransform, SinhTransform, TanhTransform, Transform
  from pymc.logprob.utils import ParameterValueError
  from pymc.testing import Rplusbig, Vector, assert_no_rvs
  from tests.distributions.test_transform import check_jacobian_det
label_desc: Use image node_id 2 for calling the TestTransform obj's underlying member method code with example usage: obj.test_fallback_log_jac_det_undefined_ndim(ndim) and returns: pt, pt