code: def testSparsityMbyNMaskingSimple(
      self, weights, expected_mask, m_by_n=(2, 4)
  ):
      weights_ts = tf.Variable(weights)
      expected_mask_ts = tf.constant(expected_mask)
  
      mask = self._sparsity_m_by_n_masking(weights_ts, m_by_n)
      self.assertAllEqual(mask, expected_mask_ts)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import compat
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_impl
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
label_desc: Use image node_id 12 for calling the PruningTest obj's underlying member method code with example usage: obj.testSparsityMbyNMaskingSimple(weights, expected_mask, m_by_n) without return types