code: def testUpdatePruningStepsAndLogsSummaries(self):
      log_dir = tempfile.mkdtemp()
      pruned_model, x_train, y_train = self._pruned_model_setup()
      pruned_model.fit(
          x_train,
          y_train,
          batch_size=self._BATCH_SIZE,
          epochs=3,
          callbacks=[
              pruning_callbacks.UpdatePruningStep(),
              pruning_callbacks.PruningSummaries(log_dir=log_dir),
          ],
      )
  
      self.assertEqual(
          3,
          tf.keras.backend.get_value(
              pruned_model.layers[0].pruning_step
          ),
      )
      self.assertEqual(
          3,
          tf.keras.backend.get_value(
              pruned_model.layers[1].pruning_step
          ),
      )
  
      self._assertLogsExist(log_dir)
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks_test.py
imports: import os
  import tempfile
  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 pruning_callbacks
label_desc: Use image node_id 3 for calling the PruneCallbacksTest obj's underlying member method code with example usage: obj.testUpdatePruningStepsAndLogsSummaries() without return types