code: def _TestRandomSmall(dtype, batch_dims, size):
      def Test(self):
          np.random.seed(42)
          shape = batch_dims + (size, size)
          matrix = np.random.uniform(
              low=-1.0, high=1.0, size=shape
          ).astype(dtype)
          self._verifyExponentialReal(matrix)
  
      return Test
  
imports: import itertools
  import numpy
  from tensorflow.python.client import session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.ops.linalg import linalg_impl
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 2 for calling a global function with example usage: _TestRandomSmall(dtype, batch_dims, size) and returns: Test