运行测试用例时，需要判断用例是否执行成功，此时需要有一个我们期望的结果来进行验证。这里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