配置
首先,你的 VS Code 要安装 Python 的插件,单元测试功能默认是关闭的,需要在 settings.json 里面添加一行:"python.testing.pytestEnabled": true。这里我使用的是第三方的 pytest 模块,你也可以使用标准库里的 unittest:"python.testing.unittestEnabled": true 或者另外一个第三方模块 nose:"python.testing.nosetestsEnabled": true。同时只能启用一个。
使用
如果你使用的是 pytest,VS Code 默认会在文件的当前目录或者子目录下查找以 test_ 开头或 _test 结尾的文件作为单元测试的脚本。这里我新建了一个文件夹,里面有两个文件:
myabs.py
def myabs(n):
if n < 0:
return -n
else:
return n
test_myabs.py
import pytest
from myabs import myabs
def test_myabs():
assert myabs(-1) == 1
assert myabs(23.5) == 23.5
接着,用 VS Code 打开这个文件夹(如果只打开文件,将不能自动识别)。等一下,你就会看到单元测试的标志:
你可以点击测试函数的 Run Test 运行单个测试,也可以点击状态栏里的 Run Tests 运行所有的测试。点击后立刻就能看到结果。