code: def testSolveComplex(self):
      # 1x1 matrix, single rhs.
      matrix = np.array([[0.1 + 1j * 0.1]])
      rhs0 = np.array([[1.0 + 1j]])
      self._verifySolveAllWaysComplex(matrix, rhs0)
      # 2x2 matrix with 2 bands, single right-hand side.
      # Corresponds to
      # [[1. + 1j, 0.], [4 + 1j, 2 + 1j]]
      matrix = np.array([[1.0, 2.0], [3.0, 4.0]]).astype(np.complex64)
      matrix += 1j * matrix
      rhs0 = np.array([[1.0], [1.0]]).astype(np.complex64)
      rhs0 += 1j * rhs0
      self._verifySolveAllWaysComplex(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]]).astype(
          np.complex64
      )
      rhs1 += 1j * rhs1
      self._verifySolveAllWaysComplex(matrix, rhs1)
  
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 7 for calling the BandedTriangularSolveOpTest obj's underlying member method code with example usage: obj.testSolveComplex() without return types