code: def testSolve(self):
      # 1x1 matrix, single rhs.
      matrix = np.array([[0.1]])
      rhs0 = np.array([[1.0]])
      self._verifySolveAllWaysReal(matrix, rhs0)
      # 2x2 matrix with 2 bands, single right-hand side.
      # Corresponds to the lower triangular
      # [[1., 0.], [3., 4.]]
      # and upper triangular
      # [[2., 1.], [0., 3.]]
      matrix = np.array([[1.0, 4.0], [2.0, 3.0]])
      rhs0 = np.array([[1.0], [1.0]])
      self._verifySolveAllWaysReal(matrix, rhs0)
      # 2x2 matrix with 2 bands, 3 right-hand sides.
      rhs1 = np.array([[1.0, 0.0, 1.0], [0.0, 1.0, 1.0]])
      self._verifySolveAllWaysReal(matrix, rhs1)
      # 4 x 4 matrix with 2 bands, 3 right hand sides.
      # Corresponds to the lower triangular
      # [[1.,  0., 0., 0.],
      #  [-1., 2., 0., 0.],
      #  [0., -2., 3., 0.],
      #  [0., 0., -3., 4.]]
      # and upper triangular
      # [[1.,  1., 0., 0.],
      #  [0., -1., 2., 0.],
      #  [0., 0., -2., 3.],
      #  [0., 0., 0., -3.]]
      matrix = np.array([[1.0, 2.0, 3.0, 4.0], [1.0, -1.0, -2.0, -3.0]])
      rhs0 = np.array(
          [
              [1.0, 0.0, 1.0],
              [0.0, 1.0, 1.0],
              [-1.0, 2.0, 1.0],
              [0.0, -1.0, -1.0],
          ]
      )
      self._verifySolveAllWaysReal(matrix, rhs0)
  
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 5 for calling the BandedTriangularSolveOpTest obj's underlying member method code with example usage: obj.testSolve() without return types