书写内存消耗程序(python)

直接上代码

#!/usr/bin/python
# -*- coding: utf-8 -*-
import sys
import re
import time

def print_help():
    print 'Usage: '
    print '  python mem.py 100MB'
    print '  python mem.py 1GB'


if __name__ == "__main__":
    if len(sys.argv) == 2:
        while True:
            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)
                elif unit == 'GB':
                    s = ' ' * (num * 1024 * 1024 * 1024)

                time.sleep(100)
                del s
            else:
                print_help()
                break
    else:
        print_help()