code: def _test_batch_stats(self, batch_expected):
      batch_actual_stats = self.tracker.get_stats(self.layer_names)
  
      self._test_stats(
          batch_expected[self.layer_1_name],
          batch_actual_stats[self.layer_1_name],
      )
  
      self._test_stats(
          batch_expected[self.layer_2_name],
          batch_actual_stats[self.layer_2_name],
      )
  
imports: import math
  from tempfile import TemporaryDirectory
  from unittest import TestCase, skipIf
  from unittest.mock import ANY, MagicMock, call
  import torch
  import torch.nn
  from tests.framework.tools import some_data_generator
  from poutyne import GradientTracker, Model, TensorBoardGradientTracker, WeightsGradientsStatsTracker
label_desc: Use image node_id 4 for calling the GradientStatsTrackerTest obj's underlying member method code with example usage: obj._test_batch_stats(batch_expected) without return types