code: def _check_mask_count(self, expected_mask_count=0):
      mask_count = 0
      for l in self.model.layers:
          mask_count += len(l.pruning_vars)
      self.assertEqual(mask_count, expected_mask_count)
  
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 3 for calling the PruningWrapperTest obj's underlying member method code with example usage: obj._check_mask_count(expected_mask_count) without return types