code: def test_returntype(self):
      a = np.array([[0, 1], [0, 0]])
      assert_(type(matrix_power(a, 2)) is np.ndarray)
      a = mat(a)
      assert_(type(matrix_power(a, 2)) is matrix)
  
imports: import collections.abc
  import numpy
  from numpy import matrix, asmatrix, bmat
  from numpy.testing import assert_, assert_equal, assert_almost_equal, assert_array_equal, assert_array_almost_equal, assert_raises
  from numpy.linalg import matrix_power
  from numpy.matrixlib import mat
label_desc: Use image node_id 1 for calling the TestPower obj's underlying member method code with example usage: obj.test_returntype() without return types