docstr: See base class.
code: def common_asserts_for_test_data(self, data):
      """See base class."""
      self.assertAllClose(data.x, data.decoded_x)
      self.assertLen(  # The Tensor rank of encoded_x should be 1.
          data.encoded_x[
              stages_impl.FlattenEncodingStage.ENCODED_VALUES_KEY
          ].shape,
          1,
      )
      self.assertAllClose(
          data.x.flatten(),
          data.encoded_x[
              stages_impl.FlattenEncodingStage.ENCODED_VALUES_KEY
          ],
      )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.stages import stages_impl
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 4 for calling the FlattenEncodingStageTest obj's underlying member method code with example usage: obj.common_asserts_for_test_data(data) without return types