code: def _verifySquareRoot(self, matrix, np_type):
      matrix = matrix.astype(np_type)
  
      # Verify that matmul(sqrtm(A), sqrtm(A)) = A
      sqrt = gen_linalg_ops.matrix_square_root(matrix)
      square = test_util.matmul_without_tf32(sqrt, sqrt)
      self.assertShapeEqual(matrix, square)
      self.assertAllClose(matrix, square, rtol=1e-4, atol=1e-3)
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import gen_linalg_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the SquareRootOpTest obj's underlying member method code with example usage: obj._verifySquareRoot(matrix, np_type) without return types