运行测试用例时，需要判断用例是否执行成功，此时需要有一个我们期望的结果来进行验证。这里unittest中，如果一个case执行的过程中报错，或者我们判断结果不符合期望，就会判定此条用例执行失败，判断的条件主要是根据断言来实现，这节主要学习下断言的使用。
一、断言的方法
1.1 TestCase类中的部分断言方法
判断条件是否成立
assert_(expr,msg)
expr:判断条件，返回Boolean值
msg:判断失败后显示的信息
判断两个数据的值是否相同
assertEqual(first,second,msg)
first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断两个数据的值是否不相同
assertEqual(first,second,msg)
first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断条件是否为真
assertTrue(expr,msg)
expr:判断条件
msg:判断失败后显示的信息
判断条件是否为假
assertFalse(expr,msg)
expr:判断条件
msg:判断失败后显示的信息
判断两个数据是否相同
assertIs(first,second,msg)
first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断两个数据是否不相同
assertIsNot(first,second,msg)
first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断数据是否存在
assertIsNone(obj,msg)
obj:所传的数据
msg:判断失败后显示的信息
判断数据是否不存在
assertIsNotNone(obj,msg)
obj:所传的数据
msg:判断失败后显示的信息
判断数据a是否包含在数据b中
assertIn(a,b,msg)
a:数据a
b:数据b
msg:判断失败后显示的信息
判断数据a是否不包含在数据b中
assertNotIn(a,b,msg)
a:数据a
b:数据b
msg:判断失败后显示的信息
1.2 直接使用assert断言
习惯直接使用python中的assert断言，替代unittest.TestCase中的断言方法：
具体可以直接使用：
assert a == b
assert a != b
assert a in b
assert a not in b
assert a in None
assert a in not None
二、在框架中使用断言
由于在使用unittest中，每个类都继承了unittest.TestCase，所以使用时，直接在方法里使用self来调用断言方法。
import unittest
class Demo(unittest.TestCase):
def test_01(self):
self.assertEqual("demo","demo","两者不相等")
如果直接使用assert断言，则可以直接调用。
import unittest
class Demo(unittest.TestCase):
def test_01(self):
assert "demo" == "demo"
三、实例演示
这里使用百度贴吧的例子，第一个case是打开新页面，然后判断新页面的title是否是我们期望的，第二个case则是我们故意判断一个错误的title，使得断言失败，这两个case也分别使用上述的两种断言方法。
# -*- coding:utf-8 -*-
from selenium import webdriver
import unittest
from time import sleep
class PcLogin(unittest.TestCase):
@classmethod
def setUpClass(cls):
print("所有测试开始")
def setUp(self):
print("测试开始")
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.driver.set_page_load_timeout(30)
self.driver.get("https://tieba.baidu.com/")
def test_01_demo(self):
"""打开新页签"""
self.driver.find_element_by_css_selector("a[title = '娱乐明星']").click()
all = self.driver.window_handles
self.driver.switch_to.window(all[1])
sleep(5)
assert self.driver.title == "娱乐明星_百度贴吧", "断言失败，当前页面title是%s " % self.driver.title
def test_02_demo2(self):
"""判断首页标题"""
# 这个例子会断言失败
self.assertEqual(self.driver.title, "娱乐明星_百度贴吧")
def tearDown(self):
print("测试完成")
self.driver.close()
@classmethod
def tearDownClass(cls):
print("所有测试结束")
if __name__ == '__main__':
unittest.main()
运行结果：（运行了两条用例，其中失败了一条；符合我们预期的结果）
所有测试开始
测试开始
测试完成
.测试开始
测试完成
F
======================================================================
FAIL: test_02_demo2 (__main__.PcLogin)
判断首页标题
所有测试结束
----------------------------------------------------------------------
Traceback (most recent call last):
File "D:/1git/uitest/test_case/demo.py", line 30, in test_02_demo2
self.assertEqual(self.driver.title, "娱乐明星_百度贴吧")
AssertionError: '百度贴吧——全球最大的中文社区' != '娱乐明星_百度贴吧'
- 百度贴吧——全球最大的中文社区
+ 娱乐明星_百度贴吧
----------------------------------------------------------------------
Ran 2 tests in 35.273s
FAILED (failures=1)
Process finished with exit code 1
