code: def testEmbedding(self):
      self.model.add(
          Prune(
              layers.Embedding(10, 10), input_shape=(10,), **self.params
          )
      )
      self.model.build(input_shape=(10, 1))
  
      self._check_mask_count(expected_mask_count=1)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 5 for calling the PruningWrapperTest obj's underlying member method code with example usage: obj.testEmbedding() without return types