你会写单元测试吗
关于我
一个有思想的程序猿,终身学习实践者,目前在一个创业团队任team lead,技术栈涉及Android、Python、Java和Go,这个也是我们团队的主要技术栈。
Github:https://github.com/hylinux1024
微信公众号:终身开发者(angrycode)
也许你已经听说过Test Driven Development
,但不知道你是否遵循这个规则呢?其实我自己在写代码的时候也很少会先写单元测试再写业务功能逻辑。这不我也今天也来学习如何在Python
中写单元测试。
0x00 unittest
Python
中的unittest
单元测试框架跟其它语言如JUnit
是类似的。它支持测试自动化、配置共享以及关机代码测试。
假设在我的项目目录下有一个mysum
模块用于计算列表中各个数之和。
还有一个test_mysum.py
用于编写单元测试的文件。
myproject/
│
├── mysum/
│ └── __init__.py
└── unittests
└── test_mysum.py
打开mysum
模块中的__init__.py
文件。
添加下面的方法
def sum(args):
total = 0
for arg in args:
total += arg
return total
打开test_mysum.py
,编写单元测试
import unittest
from mysum import sum
class TestSum(unittest.TestCase):
def test_list_int(self):
"""
测试一个整数列表的和
"""
data = [1, 2, 3]
result = sum(data)
self.assertEqual(result, 6)
if __name__ == '__main__':
unittest.main()
首先导入我们要测试的模块mysum
测试用例类通过继承unittest.TestCase
来实现,测试方法test_list_int
是以test
开头的。
在这个方法中定义了一个整型列表,执行sum
方法,然后判断执行结果是否与预期相符。
最后调用unittest.main()
来执行这个测试用例
Ran 1 test in 0.001s
OK
如果再添加一个方法
def test_list_sum(self):
data = [1, 3, 4]
result = sum(data)
self.assertEqual(result, 6)
执行后会看到如下类似信息
6 != 8
Expected :8
Actual :6
从这个输出信息可以看出期望值与实际值不相符,这时候如果我们的测试用例没有问题,那就要看看mysum
的实现逻辑了。
从这个例子中可以总结一个测试用例的过程:
- 构建输入数据
- 执行要测试模块,获取执行结果
- 与预期结果相比较,根据结果修改代码
0x01 setup/tearDown
在编写单元测试时,还可以重写父类的setup
和tearDown
方法,可以在执行测试逻辑开始前和结束时做一些处理。例如在setup
方法中可以初始化测试数据,在tearDown
方法做一些清理工作。
import unittest
class TestBasic(unittest.TestCase):
def setUp(self):
# 加载测试数据
self.app = App(database='fixtures/test_basic.json')
def test_customer_count(self):
self.assertEqual(len(self.app.customers), 100)
def test_existence_of_customer(self):
customer = self.app.get_customer(id=10)
self.assertEqual(customer.name, "Org XYZ")
self.assertEqual(customer.address, "10 Red Road, Reading")
def tearDown(self):
self.app.releaseDB()
在每个测试执行时setup
和tearDown
都会被执行一次。
0x02 pytest
pytest
是一个第三方测试框架,使用它不需要继承某个类,它可以使用原生的assert
语句用于测试结果的断言。
它的用法也很简单
首先通过pip
安装
➜ pip install pytest
我们写一个单独的tests
文件夹下创建测试用例文件test_pytest.py
。
注意:这里pytest
的文件必须与上文的unittests
文件必须区分开,否则会出现ModuleNotFoundError
。我已经在这里踩坑。
我这里使用pytest
单元测试的文件结构为
tests
│
└── test_func.py
test_func.py
的内容为
# 导入我们要测试的模块
from mysum import sum
def test_answer():
data = [1, 2, 3]
assert sum(data) == 5
测试方法以test
开头
然后再命令行中执行
➜ python -m pytest tests/test_func.py
注意:这里要使用python -m pytest
,如果直接使用pytest
会提示ModuleNotFoundError: No module named 'mysum'
执行结果如下
tests/test_func.py F [100%]
======================================================= FAILURES ==============================================
_______________________________________________________ test_answer ___________________________________________
def test_answer():
data = [1, 2, 3]
> assert sum(data) == 5
E assert 6 == 5
E + where 6 = sum([1, 2, 3])
tests/test_func.py:11: AssertionError
=======================================================1 failed in 0.03 seconds =======================================================
由于6!=5,这个单元测试提示了出错的位置。
0x03 总结一下
测试代码对编写代码非常重要,写单元测试也是一个好习惯。本文也只是一个开胃菜。要写出健壮的代码,从写单元测试开始吧。
0x04 学习资料
- https://docs.python.org/3/library/unittest.html
Unit testing framework - https://realpython.com/python-testing/
Getting Started With Testing in Python - https://docs.python-guide.org/writing/tests/
Testing Your Code
原文地址:https://www.cnblogs.com/angrycode/p/11414618.html
- Spring【AOP模块】就这么简单
- 【专业技术】如何写出优美的C 代码?
- StringUtils工具类常用方法
- 数据刷新中的并行改进(三) (r5笔记第79天)
- SSH【史上最详细整合】
- Github 项目推荐 | TensorFlow 项目模板架构最佳实践
- 两个死锁的实例 (r5笔记第90天)
- SSM【史上最详细整合】
- 终结 finalize()和垃圾回收(garbage collection)
- 代码+实战:TensorFlow Estimator of Deep CTR —— DeepFM/NFM/AFM/FNN/PNN
- 【SSH测试整合Demo】企业人事管理系统
- Hybris 项目工程配置
- 购物车案例【简单版】
- 再学习之MyBatis.
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- php 输出缓冲 Output Control用法实例详解
- PHP使用gearman进行异步的邮件或短信发送操作详解
- php多进程并发编程防止出现僵尸进程的方法分析
- php+ajax实现文件切割上传功能示例
- php操作redis数据库常见方法实例总结
- php使用pthreads v3多线程实现抓取新浪新闻信息操作示例
- PHP pthreads v3使用中的一些坑和注意点分析
- php ActiveMQ的安装与使用方法图文教程
- ThinkPHP5与单元测试PHPUnit使用详解
- php实现通过stomp协议连接ActiveMQ操作示例
- PHP pthreads v3下的Volatile简介与使用方法示例
- php实现根据身份证获取精准年龄
- php 使用ActiveMQ发送消息,与处理消息操作示例
- php使用gearman进行任务分发操作实例详解
- laravel框架select2多选插件初始化默认选中项操作示例