python+selenium2学习笔记unittest-03断言

断言的方法网上归纳的很多主要有以下这些

断言语法	解释
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()

用法还是挺简单的