From f908e3c658470d0a5581a4853e62b9d4168c3f88 Mon Sep 17 00:00:00 2001
From: Lars Wirzenius <liw@liw.fi>
Date: Wed, 16 Feb 2011 23:09:45 +0000
Subject: Add --log-memory-use option to speed-test.

---
 speed-test | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'speed-test')

diff --git a/speed-test b/speed-test
index 1e0d622..880e597 100755
--- a/speed-test
+++ b/speed-test
@@ -28,6 +28,7 @@
 
 import cliapp
 import cProfile
+import logging
 import os
 import random
 import shutil
@@ -146,6 +147,8 @@ class SpeedTest(cliapp.Application):
         def helper():
             for item in items:
                 func(item)
+            if self['log-memory-use']:
+                logging.debug('VmRSS: %s KiB' % self.vmrss())
             finalize()
 
         print 'measuring', profname
@@ -162,6 +165,15 @@ class SpeedTest(cliapp.Application):
         end_time = time.time()
         return end - start, end_time - start_time
 
+    def vmrss(self):
+        f = open('/proc/self/status')
+        rss = 0
+        for line in f:
+            if line.startswith('VmRSS'):
+                rss = line.split()[1]
+        f.close()
+        return rss
+
 
 if __name__ == '__main__':
     SpeedTest().run()
-- 
cgit v1.2.1