code: def testWrongDimensions(self):
      # The matrix should have the same number of rows as the
      # right-hand sides.
      matrix = np.array([[1.0, 1.0], [1.0, 1.0]])
      rhs = np.array([[1.0, 0.0]])
      with self.cached_session():
          with self.assertRaises(ValueError):
              self._verifySolve(matrix, rhs)
          with self.assertRaises(ValueError):
              self._verifySolve(matrix, rhs, batch_dims=[2, 3])
  
      # Number of bands exceeds the dimension of the matrix.
      matrix = np.ones((6, 4))
      rhs = np.ones((4, 2))
      with self.cached_session():
          with self.assertRaises(ValueError):
              self._verifySolve(matrix, rhs)
          with self.assertRaises(ValueError):
              self._verifySolve(matrix, rhs, batch_dims=[2, 3])
  
file_path: tensorflow-upstream/tensorflow/python/kernel_tests/math_ops/banded_triangular_solve_op_test.py
imports: import numpy
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import linalg_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 10 for calling the BandedTriangularSolveOpTest obj's underlying member method code with example usage: obj.testWrongDimensions() without return types