用 VS Code 进行 Python 单元测试

配置

首先，你的 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 运行所有的测试。点击后立刻就能看到结果。

参考资料

https://code.visualstudio.com/docs/python/testing