code: def test_collected_stats(self):
      x = torch.randn(32, 128)
      d = LpDistance()
      d(x)
  
      self.assertNotEqual(d.initial_avg_query_norm, 0)
      self.assertNotEqual(d.initial_avg_ref_norm, 0)
      self.assertNotEqual(d.final_avg_query_norm, 0)
      self.assertNotEqual(d.final_avg_ref_norm, 0)
  
imports: import unittest
  import torch
  from pytorch_metric_learning.distances import LpDistance
  from ..None import WITH_COLLECT_STATS
label_desc: Use image node_id 1 for calling the TestCollectedStats obj's underlying member method code with example usage: obj.test_collected_stats() without return types