code: def test_train_and_test_weights_are_equal(self, *shape):
      algorithm = self.get_algorithm()
      init = tf.random.uniform(shape, dtype=tf.float32)
      algorithm.init_training_weights(init)
      layer = tf.keras.layers.Layer()
      for weight_repr in algorithm.weight_reprs:
          layer.add_weight(*weight_repr.args, **weight_repr.kwargs)
      train_weight = algorithm.project_training_weights(*layer.weights)
      compressed_weights = algorithm.compress_training_weights(
          *layer.weights
      )
      test_weight = algorithm.decompress_weights(*compressed_weights)
      self.assertAllEqual(train_weight, test_weight)
  
imports: import os
  import shutil
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.common.keras.compression.algorithms import epr
label_desc: Use image node_id 6 for calling the EPRTest obj's underlying member method code with example usage: obj.test_train_and_test_weights_are_equal() without return types