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

背景知识

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

官方文档：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方法，可以运行全部的测试案例，也可以只运行一个测试用例。