code: def _check_strip_pruning_matches_original(
      self, model, sparsity, input_data=None
  ):
      test_utils.assert_model_sparsity(self, sparsity, model)
      stripped_model = prune.strip_pruning(model)
  
      if input_data is None:
          input_data = np.random.randn(
              *self._batch(model.input.get_shape().as_list(), 1)
          )
  
      model_result = model.predict(input_data)
      stripped_model_result = stripped_model.predict(input_data)
      np.testing.assert_almost_equal(
          model_result, stripped_model_result
      )
  
imports: import tempfile
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import test_utils
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import prune_registry
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_callbacks
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
  from tensorflow_model_optimization.python.core.sparsity.keras import test_utils
label_desc: Use image node_id 4 for calling the PruneIntegrationTest obj's underlying member method code with example usage: obj._check_strip_pruning_matches_original(model, sparsity, input_data) without return types