Python项目中的单元测试

李從新 發表於 2019-8-16 16:50:00

<h1>引入</h1>
单元测试负责对最小的软件设计单元（模块）进行验证,<code>unittest</code>是Python自带的单元测试框架。单元测试与功能测试都是日常开发中必不可少的部分，本文演示了Python中<code>unittest</code>单元测试框架的基本使用。
 
<h1>一个简单的测试例子</h1>
定义一个类，简单的实现<code>add</code>、<code>sub</code>两个方法，并对其进行单元测试。
待测试的<code>m1.py</code>文件内容如下：
<div class="cnblogs_code">
<pre>class MyClass(object):
"""just a test case"""
def __init__(self, x, y):
 self.x = int(x)
 self.y = int(y)

def add(self):
 return self.x + self.y

def sub(self):
 return self.x - self.y</pre>
</div>
在与<code>m1.py</code>同级的目录下创建<code>test.py</code>测试文件，使用<code>unittest</code>单元测试框架对A类的方法进行测试。代码内容如下：
<div class="cnblogs_code">
<pre>import unittest
from m1 import MyClass

class MyClassTest(unittest.TestCase):
def setUp(self):
 self.calc = MyClass(7, 5)

def tearDown(self):
 pass

def test_add(self):
 ret = self.calc.add()
 self.assertEqual(ret, 12)

def test_sub(self):
 ret = self.calc.sub()
 self.assertEqual(ret, 2)

if __name__ == '__main__':
# 构造测试集
suite = unittest.TestSuite()
suite.addTest(MyClassTest('test_add'))
suite.addTest(MyClassTest('test_sub'))
# 执行测试
runner = unittest.TextTestRunner()
runner.run(suite)</pre>
</div>
运行测试：
<div class="cnblogs_code">
<pre>demo1 $ python3 test.py
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s

OK</pre>
</div>
到此一个简单的单元测试就完成了。
 
<h1 id="autoid-1-0-0">unittest框架知识点</h1>
unittest框架中4个重要的概念：
<ol>
<li><code>test fixture：</code>是初始化和清理测试数据及环境，通过重写TestCase的<code>setUp()</code>和<code>tearDown()</code>方法来实现</li>
<li><code>test case：</code>是测试用例</li>
<li><code>test suite：</code>是测试用例的集合（俗称测试套件），通过addTest加载TestCase到TestSuite中，返回一个TestSuite实例。</li>
<li><code>test runner：</code>的作用是运行测试用例并返回结果，通过TextTestRunner类提供的run()方法来执行<code>test suite</code>或<code>test case</code>。</li>
</ol>
 
 
<h1 id="autoid-2-0-0">Django中的单元测试</h1>
Django项目的app目录下都默认生成了一个
<h2 id="autoid-3-0-0">Model部分单元测试用例</h2>
假设项目中有一个<code>Book</code>的model：
<div class="cnblogs_code">
<pre>class Book(models.Model):
title = models.CharField(max_length=32)
price = models.DecimalField(max_digits=10, decimal_places=2)</pre>
</div>
测试用例代码：
<div class="cnblogs_code">
<pre># app01/tests.py
from django.test import TestCase
from app01.models import Book
# Create your tests here.

class BookModelTest(TestCase):
def setUp(self):
 Book.objects.create(title='书名', price=11.11)

def test_book_model(self):
 from decimal import Decimal
 result = Book.objects.get(title='书名')
 self.assertEqual(result.price, Decimal('11.11'))</pre>
</div>
运行测试，在项目目录下运行：
<div class="cnblogs_code">
<pre>$ python3 manage.py test
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
.
----------------------------------------------------------------------
Ran 1 test in 0.003s

OK
Destroying test database for alias 'default'...</pre>
</div>
测试用例OK…
<h2 id="autoid-3-1-0">视图部分单元测试用例</h2>
假设我们有个<code>index</code>视图，代码如下：
<div class="cnblogs_code">
<pre>def index(request):
return render(request, 'index.html')</pre>
</div>
在<code>app01/tests.py</code>文件中添加测试用例代码：
<div class="cnblogs_code">
<pre># app01/tests.py
class IndexPageTest(TestCase):
"""测试index页面"""
def test_index_page_renders_index_template(self):
 """测试index视图"""
 response = self.client.get('/index/')
 self.assertEqual(response.status_code, 200)# 判断状态码
 self.assertTemplateUsed(response, 'index.html')# 判断渲染的模板是否正确</pre>
</div>
在项目根目录运行<code>python manage.py test</code>命令：
<div class="cnblogs_code">
<pre>$ python3 manage.py test
Creating test database for alias 'default'...
System check identified no issues (0 silenced).
..
----------------------------------------------------------------------
Ran 2 tests in 0.027s

OK
Destroying test database for alias 'default'...</pre>
</div>
<code>python manage.py test</code>命令会查找当前项目中的tests.py文件，并运行测试用例。 
来源：https://www.cnblogs.com/peng104/p/11364927.html

頁: [1]

琼殿技术论坛's Archiver

Python项目中的单元测试