code: def _assertNdSquareness(*arrays):
      for a in arrays:
          m, n = a.shape[-2:]
          if m != n:
              raise np.linalg.LinAlgError(
                  "Last 2 dimensions of the array must be square"
              )
  
file_path: jax/jax/_src/third_party/numpy/linalg.py
imports: import numpy
  import jax.numpy
  import jax.numpy.linalg
  from jax._src.numpy.util import check_arraylike, _wraps
label_desc: Use image node_id 4 for calling a global function with example usage: _assertNdSquareness() without return types