Python 代码测试

2024-06-15 | 阅读：次

本文记录了在 Python 代码性能或者运行结果不符合预期时应该如何测试.

把这个放在前面是因为感觉比较容易理解.

众所周知, 代码报错的时候 Python 会给出程序栈中所有的调用流程, 可以清晰的看到程序是如何到达报错位置的, 而我们可以手动输出栈中内容:

import traceback

# 在想要查看的地方安置这段代码即可
for line in traceback.format_stack():
    print(line, end="")

首先生成运行监控文件:

from cProfile import Profile
p = Profile()
p.enable()
... # 要测试的内容
p.disable()
p.dump_stats("profile.dump")

之后使用 snakeviz 生成浏览页面:

pip install snakeviz
snakeviz profile.dump
# or
python -m snakeviz profile.dump output.html

pip install austin-tui
python -m program
austin-tui -p <PID>
# or
austin-tui python -m program

Windows 端如果报错无法启动或者找不到 AustinError 可以手动安装一下 austin 选择新版本下载 msi 文件, 直接安装即可.

H_On