断言的方法网上归纳的很多主要有以下这些
断言语法 | 解释 |
assertEqual(a, b) | 判断a==b |
assertNotEqual(a, b) | 判断a!=b |
assertTrue(x) | bool(x) is True |
assertFalse(x) | bool(x) is False |
assertIs(a, b) | a is b |
assertIsNot(a, b) | a is not b |
assertIsNone(x) | x is None |
assertIsNotNone(x) | x is not None |
assertIn(a, b) | a in b |
assertNotIn(a, b) | a not in b |
assertIsInstance(a, b) | isinstance(a, b) |
assertNotIsInstance(a, b) | not isinstance(a, b) |
我们看例子
import unittest
class test(unittest.TestCase):
def setUp(self):
number = input("输入数字")
self.number = int(number)
def test_case(self):
self.assertEqual(self.number,10,msg="输入的不是10")
def tearDown(self):
pass
if __name__ == "__main__":
unittest.main()
再看一个判断素数的例子
#判断素数(质数) 只能被1或本身整除的数
def is_prime(n):
if n <= 1:
return False
for i in range(2,n):
if n % i == 0:
return False
return True
from selenium import webdriver
import unittest
from 断言素数 import is_prime
class test(unittest.TestCase):
def setUp(self):
print("start")
def test_case(self):
self.assertTrue(is_prime(6),msg="不是素数")
def tearDown(self):
pass
if __name__ == "__main__":
unittest.main()
用法还是挺简单的