class SnapshotTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.stop_periodic_snapshots()
self.tracker.clear()
def test_timestamp(self):
"""Test timestamp of snapshots.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo)
self.tracker.track_object(bar)
self.tracker.create_snapshot()
self.tracker.create_snapshot()
self.tracker.create_snapshot()
refts = [fp.timestamp for fp in self.tracker.snapshots]
for to in self.tracker.objects.values():
ts = [t for (t,sz) in to.snapshots[1:]]
self.assertEqual(ts,refts)
def test_snapshot_members(self):
"""Test existence and value of snapshot members.
"""
foo = Foo()
self.tracker.track_object(foo)
self.tracker.create_snapshot()
self.tracker.create_snapshot(compute_total=True)
fp = self.tracker.snapshots[0]
fp_with_total = self.tracker.snapshots[1]
self.assert_(fp.overhead > 0, fp.overhead)
self.assert_(fp.tracked_total > 0, fp.tracked_total)
self.assertEqual(fp.asizeof_total, 0)
self.assert_(fp_with_total.asizeof_total > 0, fp_with_total.asizeof_total)
self.assert_(fp_with_total.asizeof_total >= fp_with_total.tracked_total)
if pympler.process.is_available():
procmem = fp.system_total
self.assertEqual(fp.total, procmem.vsz)
self.assert_(procmem.vsz > 0, procmem)
self.assert_(procmem.rss > 0, procmem)
self.assertTrue(procmem.vsz >= procmem.rss, procmem)
self.assert_(procmem.vsz > fp.overhead, procmem)
self.assert_(procmem.vsz > fp.tracked_total, procmem)
self.assert_(fp_with_total.system_total.vsz > fp_with_total.asizeof_total)
else:
self.assertEqual(fp_with_total.total, fp_with_total.asizeof_total)
self.assertEqual(fp.total, fp.tracked_total)
def test_desc(self):
"""Test snapshot label.
"""
self.tracker.create_snapshot()
self.tracker.create_snapshot('alpha')
self.tracker.create_snapshot(description='beta')
self.tracker.create_snapshot(42)
self.assertEqual(len(self.tracker.snapshots), 4)
self.assertEqual(self.tracker.snapshots[0].desc, '')
self.assertEqual(self.tracker.snapshots[1].desc, 'alpha')
self.assertEqual(self.tracker.snapshots[2].desc, 'beta')
self.assertEqual(self.tracker.snapshots[3].desc, '42')
snapshot = self.tracker.snapshots[0]
self.assertEqual(snapshot.label, '%.3fs' % snapshot.timestamp)
snapshot = self.tracker.snapshots[1]
self.assertEqual(snapshot.label, 'alpha (%.3fs)' % snapshot.timestamp)
snapshot = self.tracker.snapshots[3]
self.assertEqual(snapshot.label, '42 (%.3fs)' % snapshot.timestamp)
def test_background_monitoring(self):
"""Test background monitoring.
"""
self.tracker.start_periodic_snapshots(0.1)
self.assertEqual(self.tracker._periodic_thread.interval, 0.1)
self.assertEqual(self.tracker._periodic_thread.getName(), 'BackgroundMonitor')
for x in range(10): # try to interfere
self.tracker.create_snapshot(str(x))
time.sleep(0.5)
self.tracker.start_periodic_snapshots(0.2)
self.assertEqual(self.tracker._periodic_thread.interval, 0.2)
self.tracker.stop_periodic_snapshots()
self.assert_(self.tracker._periodic_thread is None)
self.assert_(len(self.tracker.snapshots) > 10)
class SnapshotTestCase(unittest.TestCase):
def setUp(self):
self.tracker = ClassTracker()
def tearDown(self):
self.tracker.stop_periodic_snapshots()
self.tracker.clear()
def test_timestamp(self):
"""Test timestamp of snapshots.
"""
foo = Foo()
bar = Bar()
self.tracker.track_object(foo)
self.tracker.track_object(bar)
self.tracker.create_snapshot()
self.tracker.create_snapshot()
self.tracker.create_snapshot()
refts = [fp.timestamp for fp in self.tracker.snapshots]
for to in self.tracker.objects.values():
ts = [t for (t, sz) in to.snapshots[1:]]
self.assertEqual(ts, refts)
def test_snapshot_members(self):
"""Test existence and value of snapshot members.
"""
foo = Foo()
self.tracker.track_object(foo)
self.tracker.create_snapshot()
self.tracker.create_snapshot(compute_total=True)
fp = self.tracker.snapshots[0]
fp_with_total = self.tracker.snapshots[1]
self.assertTrue(fp.overhead > 0, fp.overhead)
self.assertTrue(fp.tracked_total > 0, fp.tracked_total)
self.assertEqual(fp.asizeof_total, 0)
self.assertTrue(fp_with_total.asizeof_total > 0,
fp_with_total.asizeof_total)
self.assertTrue(
fp_with_total.asizeof_total >= fp_with_total.tracked_total)
if pympler.process.is_available():
procmem = fp.system_total
self.assertEqual(fp.total, procmem.vsz)
self.assertTrue(procmem.vsz > 0, procmem)
self.assertTrue(procmem.rss > 0, procmem)
self.assertTrue(procmem.vsz >= procmem.rss, procmem)
self.assertTrue(procmem.vsz > fp.overhead, procmem)
self.assertTrue(procmem.vsz > fp.tracked_total, procmem)
self.assertTrue(
fp_with_total.system_total.vsz > fp_with_total.asizeof_total)
else:
self.assertEqual(fp_with_total.total, fp_with_total.asizeof_total)
self.assertEqual(fp.total, fp.tracked_total)
def test_desc(self):
"""Test snapshot label.
"""
self.tracker.create_snapshot()
self.tracker.create_snapshot('alpha')
self.tracker.create_snapshot(description='beta')
self.tracker.create_snapshot(42)
self.assertEqual(len(self.tracker.snapshots), 4)
self.assertEqual(self.tracker.snapshots[0].desc, '')
self.assertEqual(self.tracker.snapshots[1].desc, 'alpha')
self.assertEqual(self.tracker.snapshots[2].desc, 'beta')
self.assertEqual(self.tracker.snapshots[3].desc, '42')
snapshot = self.tracker.snapshots[0]
self.assertEqual(snapshot.label, '%.3fs' % snapshot.timestamp)
snapshot = self.tracker.snapshots[1]
self.assertEqual(snapshot.label, 'alpha (%.3fs)' % snapshot.timestamp)
snapshot = self.tracker.snapshots[3]
self.assertEqual(snapshot.label, '42 (%.3fs)' % snapshot.timestamp)
def test_background_monitoring(self):
"""Test background monitoring.
"""
self.tracker.start_periodic_snapshots(0.1)
self.assertEqual(self.tracker._periodic_thread.interval, 0.1)
self.assertEqual(self.tracker._periodic_thread.getName(),
'BackgroundMonitor')
for x in range(10): # try to interfere
self.tracker.create_snapshot(str(x))
time.sleep(0.5)
self.tracker.start_periodic_snapshots(0.2)
self.assertEqual(self.tracker._periodic_thread.interval, 0.2)
self.tracker.stop_periodic_snapshots()
self.assertTrue(self.tracker._periodic_thread is None)
self.assertTrue(len(self.tracker.snapshots) > 10)
