test/debug_internal.py

   1 import gc
   2 import pprint
   3 import vpp_papi
   4 from vpp_papi_provider import VppPapiProvider
   5 import objgraph
   6 from pympler import tracker
   7 tr = tracker.SummaryTracker()
   8
   9 """
  10   Internal debug module
  11
  12   The module provides functions for debugging test framework
  13 """
  14
  15
  16 def on_tear_down_class(cls):
  17     gc.collect()
  18     tr.print_diff()
  19     objects = gc.get_objects()
  20     counter = 0
  21     with open(cls.tempdir + '/python_objects.txt', 'w') as f:
  22         interesting = [
  23             o for o in objects
  24             if isinstance(o, (VppPapiProvider, vpp_papi.VPP))]
  25         del objects
  26         gc.collect()
  27         for o in interesting:
  28             objgraph.show_backrefs([o], max_depth=5,
  29                                    filename="%s/%s.png" %
  30                                    (cls.tempdir, counter))
  31             counter += 1
  32             refs = gc.get_referrers(o)
  33             pp = pprint.PrettyPrinter(indent=2)
  34             f.write("%s\n" % pp.pformat(o))
  35             for r in refs:
  36                 try:
  37                     f.write("%s\n" % pp.pformat(r))
  38                 except:
  39                     f.write("%s\n" % type(r))