Python小工具之消耗系统指定大小内存的方法

#!/usr/bin/python# -*- coding: utf-8 -*-import sysimport reimport timedef print_help():  print 'Usage: '  print ' python mem.py 100MB'  print ' python mem.py 1GB'if __name__ == "__main__":  if len(sys.argv) == 2:    pattern = re.compile('^(/d*)([M|G]B)$')    match = pattern.match(sys.argv[1].upper())    if match:      num = int(match.group(1))      unit = match.group(2)      if unit == 'MB':        s = ' ' * (num * 1024 * 1024)      else:        s = ' ' * (num * 1024 * 1024 * 1024)      time.sleep(10000)    else:      print_help()  else:    print_help()

python mem.py 100Mpython mem.py 1G