code: def assertStateEqual(self, state1, state2):
      for s1, s2 in zip(
          tf.nest.flatten(state1), tf.nest.flatten(state2)
      ):
          self.assertAllEqual(s1, s2)
  
imports: import math
  import random
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_privacy.privacy.dp_query import tree_aggregation
label_desc: Use image node_id 1 for calling the GaussianNoiseGeneratorTest obj's underlying member method code with example usage: obj.assertStateEqual(state1, state2) without return types