code: def testPruneWrapperAllowsOnlyValidPoolingType(self):
      layer = layers.Dense(10)
      with self.assertRaises(ValueError):
          pruning_wrapper.PruneLowMagnitude(
              layer, block_pooling_type="MIN"
          )
  
      pruning_wrapper.PruneLowMagnitude(layer, block_pooling_type="AVG")
      pruning_wrapper.PruneLowMagnitude(layer, block_pooling_type="MAX")
  
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 2 for calling the PruningWrapperTest obj's underlying member method code with example usage: obj.testPruneWrapperAllowsOnlyValidPoolingType() without return types