使用VS Code进行Python单元测试 提高开发效率

共计 775 个字符，预计需要花费 2 分钟才能阅读完成。

背景知识

单元测试在开发过程中非常有必要，它可以验证实现的一个函数是否达到预期。使用单元测试能提高开发效率，也更规范。

官方文档：https://code.visualstudio.com/docs/python/testing

单元测试

在 VS Code 中，可以实现对编写的每一个函数进行单元测试，理应每实现一个函数都有一个单元测试的用例。

演示

1. 配置测试文件

点击 VS Code 左边栏的测试，配置 Python 测试文件，依次选择：Configure Python Tests -> unittest -> . Root directory -> test*.py。

这里表示使用 unittest 框架测试，测试文件的格式为 test*.py。

随后在项目目录中多了一个.vscode 文件夹，其中包含一个 settings.json 文件，在这里可以设置项目的一些配置。

2. 新建一个 Python 项目 TestNumpy.py

测试文件的格式为 test*.py。

import numpy as np
from unittest import TestCase

class TestNumpy(TestCase):
    def testReadFile(self):
        file_name = "./demo.csv"
        end_price, volumn = np.loadtxt(fname=file_name, delimiter=",", usecols=(2, 6), unpack=True)
        print(end_price)
        print(volumn)

在 TestNumpy.py 中编写单元测试用例，需要引入 unittest 模块，类要继承 unittest.TestCase，并以 test 开头命名类和方法。

3. 在测试中运行单元测试用例

点击 VS Code 左边栏的测试，可以看到控制台中已经包含了 unittest 方法，可以运行全部的测试案例，也可以只运行一个测试用例。