Python实现指定内存占用的测试脚本
发布时间：2026-10-13| 4人浏览过
本文介绍了一个Python小工具，用于模拟应用程序消耗指定大小的系统内存。通过命令行参数指定内存大小（如MB或GB），脚本会分配相应内存并维持占用状态，适用于测试场景下的资源监控与压力测试参考。
在实际开发中，有时需要依据特定应用程序所占用的内存量来触发某些操作逻辑。为此，编写了一个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:
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 100M python mem.py 1G
以上就是本文关于Python小工具实现指定内存消耗的全部内容，希望能为读者提供有价值的参考，同时也欢迎对本站内容给予支持。
免责声明：本文内容仅供参考，不代表本站观点。转载或引用请注明出处。
