code: def assertAllEqual(self, a, b, msg=None):
      if not (
          isinstance(a, structured_tensor.StructuredTensor)
          or isinstance(b, structured_tensor.StructuredTensor)
      ):
          return super(StructuredArrayOpsTest, self).assertAllEqual(
              a, b, msg
          )
  
      if not isinstance(a, structured_tensor.StructuredTensor):
          a = structured_tensor.StructuredTensor.from_pyval(a)
      elif not isinstance(b, structured_tensor.StructuredTensor):
          b = structured_tensor.StructuredTensor.from_pyval(b)
  
      try:
          nest.assert_same_structure(a, b, expand_composites=True)
      except (TypeError, ValueError) as e:
          self.assertIsNone(e, (msg + ": " if msg else "") + str(e))
      a_tensors = [
          x
          for x in nest.flatten(a, expand_composites=True)
          if isinstance(x, tensor.Tensor)
      ]
      b_tensors = [
          x
          for x in nest.flatten(b, expand_composites=True)
          if isinstance(x, tensor.Tensor)
      ]
      self.assertLen(a_tensors, len(b_tensors))
      a_arrays, b_arrays = self.evaluate((a_tensors, b_tensors))
      for a_array, b_array in zip(a_arrays, b_arrays):
          self.assertAllEqual(a_array, b_array, msg)
  
imports: from absl.testing import parameterized
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import random_seed
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops.ragged import ragged_factory_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.ops.ragged import row_partition
  from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
  from tensorflow.python.ops.structured import structured_array_ops
  from tensorflow.python.ops.structured import structured_tensor
  from tensorflow.python.ops.structured.structured_tensor import StructuredTensor
  from tensorflow.python.platform import googletest
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 for calling the StructuredArrayOpsTest obj's underlying member method code with example usage: obj.assertAllEqual(a, b, msg) and returns: super