Changeset 1974

Timestamp:

Oct 25, 2005, 1:15:05 PM (19 years ago)

Author:

duncan

Message:

adding memory stuff

File:

• inundation/fit_interpolate/benchmark_least_squares.py (modified) (2 diffs)

inundation/fit_interpolate/benchmark_least_squares.py

@@ -24 +24 @@
 from Numeric import allclose, array, transpose
 import os
-
-def report_memory(i):
-    pid = os.getpid()
-    a2 = os.popen('ps -p %d -o rss,sz' % pid).readlines()
-    print i, '  ', a2[1],
-    return int(a2[1].split()[1])
-
-
-def showmem(): os.system("ps axm | head -1 ; ps axm | grep py | grep -v grep")
-
-def mem_usage(pid):
+import sys
+
+def mem_usage():
     '''
     returns the rss.
 
+  RSS  The total amount of physical memory used by  the  task,  in  kilo-
+            bytes,  is  shown  here.  For ELF processes used library pages are
+            counted here, for a.out processes not.
+            
     Only works on nix systems.
     '''
     import string
-    p=os.popen('ps uwp %s'%pid) 
+    p=os.popen('ps uwp %s'%os.getpid()) 
     lines=p.readlines()
     #print "lines", lines
     status=p.close() 
-    if status or len(lines)!=2: 
+    if status or len(lines)!=2 or sys.platform == 'win32': 
         return None 
     return int(string.split(lines[1])[4]) 
@@ -212 +208 @@
 if __name__ == "__main__":
     import os
-    #val = report_memory(1)
-    show_me = 'ps v %d' % os.getpid()
-    os.system(show_me)
     suite = unittest.makeSuite(Test_Least_Squares,'test')
     runner = unittest.TextTestRunner(verbosity=1)
-    #runner.run(suite)
-    os.system(show_me)
-    #end = report_memory(2)
-    val = mem_usage(os.getpid())
-    print 'val', val
+    runner.run(suite)
+    val = mem_usage()
+    print 'RSS', val