Selenium实战（八）——pytest单元测试（2）

一、断言

　　不像在unittest中的断言方法：assertEqual()、assertIn()、assertTrue()、assertIs()等。pytest单元测试框架并没有提供专门的断言方法，而是直接使用Python的assert进行断言。

　　创建test_assert.py文件

 1 # 功能：用于计算a与b相加的和
 2 def add(a, b):
 3     return a + b
 4 
 5 
 6 # 功能：判断素数
 7 def is_prime(n):
 8     if n <= 1:
 9         return False
10     for i in range(2, n):
11         if n % i == 0:
12             return False
13     return True
14 
15 
16 # 测试相等
17 def test_add_1():
18     assert add(3, 10) == 13
19 
20 
21 # 测试不相等
22 def test_add_2():
23     assert add(4, 8) != 98
24 
25 
26 # 测试大于或等于
27 def test_add_3():
28     assert add(9, 3) >= 11
29 
30 
31 # 小于或等于
32 def test_add_4():
33     assert add(4, 7) <= 17
34 
35 
36 # 测试包含
37 def test_in():
38     a = "hello"
39     b = "ell"
40     assert b in a
41 
42 
43 # 测试不包含
44 def test_not_in():
45     a = "print"
46     b = "OOO"
47     assert b not in a
48 
49 
50 # 判断是否为True
51 def test_true_1():
52      assert is_prime(13)
53 
54 
55 # 判断是否为True
56 def test_true_2():
57     assert is_prime(7) is True
58 
59 
60 # 判断是否不为True
61 def test_true_3():
62     assert not is_prime(4)
63 
64 
65 # 判断是否不为True
66 def test_true_4():
67     assert is_prime(6) is not True
68 
69 
70 # 判断是否为False
71 def test_false_1():
72     assert is_prime(8) is False