下面是一个 Python的单元测试简单的例子：
假如我们开发一个除法的功能，有的同学可能觉得很简单，代码是这样的：
def division_funtion(x, y): return x / y |
但是这样写究竟对还是不对呢，有些同学可以在代码下面这样测试：
def division_funtion(x, y): return x / y if __name__ == '__main__': print division_funtion(2, 1) print division_funtion(2, 4) print division_funtion(8, 3) |
但是这样运行后得到的结果，自己每次都得算一下去核对一遍，很不方便，Python中有 unittest 模块，可以很方便地进行测试，详情可以文章最后的链接，看官网文档的详细介绍。
下面是一个简单的示例：
import unittest def division_funtion(x, y): return x / y class TestDivision(unittest.TestCase): def test_int(self): self.assertEqual(division_funtion(9, 3), 3) def test_int2(self): self.assertEqual(division_funtion(9, 4), 2.25) def test_float(self): self.assertEqual(division_funtion(4.2, 3), 1.4) if __name__ == '__main__': unittest.main() |
我简单地写了三个测试示例（不一定全面，只是示范，比如没有考虑除数是0的情况），运行后发现：
FAIL: test_float (__main__.TestDivision) ---------------------------------------------------------------------- Traceback (most recent call last): File "/Users/tu/YunPan/mydivision.py", line 16, in test_float self.assertEqual(division_funtion(4.2, 3), 1.4) AssertionError: 1.4000000000000001 != 1.4 ====================================================================== FAIL: test_int2 (__main__.TestDivision) ---------------------------------------------------------------------- Traceback (most recent call last): File "/Users/tu/YunPan/1.py", line 13, in test_int2 self.assertEqual(division_funtion(9, 4), 2.25) AssertionError: 2 != 2.25 ---------------------------------------------------------------------- Ran 3 tests in 0.001s FAILED (failures=2) |
汗！发现了没，竟然两个都失败了，测试发现：
4.2除以3 等于 1.4000000000000001 不等于期望值 1.4
9除以4等于2，不等于期望的 2.25
下面我们就是要修复这些问题，再次运行测试，直到运行不报错为止。
譬如根据实际情况，假设我们只需要保留到小数点后6位，可以这样改：
def division_funtion(x, y): return round(float(x) / y, 6) |
再次运行就不报错了:
Ran 3 tests in 0.000s OK Python 单元测试 官方文档： Python 2 (https://docs.python.org/2/library/unittest.html) Python 3 (https://docs.python.org/3/library/unittest.html) |
2. Django 中 单元测试：(不断完善中，后期会增加对前面讲解的内容的测试）
2.1 简单测试例子：
from django.test import TestCase from myapp.models import Animal class AnimalTestCase(TestCase): def setUp(self): Animal.objects.create(name="lion", sound="roar") Animal.objects.create(name="cat", sound="meow") def test_animals_can_speak(self): """Animals that can speak are correctly identified""" lion = Animal.objects.get(name="lion") cat = Animal.objects.get(name="cat") self.assertEqual(lion.speak(), 'The lion says "roar"') self.assertEqual(cat.speak(), 'The cat says "meow"') |
这个例子是测试myapp.models 中的 Animal 类相关的方法功能。
2.2 用代码访问网址的方法：
>>> from django.test import Client >>> c = Client() >>> response = c.post('/login/', {'username': 'john', 'password': 'smith'}) >>> response.status_code 200 >>> response = c.get('/customer/details/') >>> response.content '<!DOCTYPE html...' |
我们可以用 django.test.Client 的实例来实现 get 或 post 内容，检查一个网址返回的网页源代码。
默认情况下CSRF检查是被禁用的，如果测试需要，可以用下面的方法：
>>> from django.test import Client >>> csrf_client = Client(enforce_csrf_checks=True) |
使用 csrf_client 这个实例进行请求即可。
指定浏览USER-AGENT:
| >>> c = Client(HTTP_USER_AGENT='Mozilla/5.0') |
模拟post上传附件：
from django.test import Client c = Client() with open('wishlist.doc') as fp: c.post('/customers/wishes/', {'name': 'fred', 'attachment': fp}) |
测试网页返回状态：
from django.test import TestCase class SimpleTest(TestCase): def test_details(self): response = self.client.get('/customer/details/') self.assertEqual(response.status_code, 200) def test_index(self): response = self.client.get('/customer/index/') self.assertEqual(response.status_code, 200) |
我们用 self.client 即可，不用 client = Client() 这样实例化，更方便，我们还可以继承 Client，添加一些其它方法:
from django.test import TestCase, Client class MyTestClient(Client): # Specialized methods for your environment ... class MyTest(TestCase): client_class = MyTestClient def test_my_stuff(self): # Here self.client is an instance of MyTestClient... call_some_test_code() |
定制 self.client 的方法：
from django.test import Client, TestCase class MyAppTests(TestCase): def setUp(self): super(MyAppTests, self).setUp() self.client = Client(enforce_csrf_checks=True) def test_home(self): response = self.client.get('/') self.assertEqual(response.status_code, 200) |
上文内容不用于商业目的，如涉及知识产权问题，请权利人联系博为峰小编(021-64471599-8017)，我们将立即处理。
