code: def testWeightsRearrange(self, shape, expected_shape):
      weights = tf.Variable(
          glorot_uniform_initializer()(shape),
          shape=shape,
          name="weights",
      )
      prepared_wt = pruning_utils.weights_rearrange(weights)
      self.assertEqual(prepared_wt.shape, expected_shape)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import compat
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_utils
label_desc: Use image node_id 1 for calling the WeightsRearrangeTest obj's underlying member method code with example usage: obj.testWeightsRearrange(shape, expected_shape) without return types