unittest断言方法代码示例

unittest是Python中常用的单元测试框架，它提供了丰富的断言方法用于判断测试结果是否符合预期。下面是一些unittest断言方法的代码示例：

1. assertEqual()
该方法用于判断两个值是否相等。

import unittest

class MyTest(unittest.TestCase):
    def test_equal(self):
        self.assertEqual(1+1, 2)  # 断言1+1等于2

2. assertNotEqual()
该方法用于判断两个值是否不相等。

import unittest

class MyTest(unittest.TestCase):
    def test_not_equal(self):
        self.assertNotEqual(1+1, 3)  # 断言1+1不等于3

3. assertTrue()和assertFalse()
这两个方法分别用于判断表达式是否为True和False。

import unittest

class MyTest(unittest.TestCase):
    def test_true(self):
        self.assertTrue(2 > 1)  # 断言2大于1为True

    def test_false(self):
        self.assertFalse(1 > 2)  # 断言1大于2为False

4. assertIn()和assertNotIn()
这两个方法用于判断一个值是否在列表或元组中存在或不存在。

import unittest

class MyTest(unittest.TestCase):
    def test_in(self):
        myList = [1, 2, 3, 4, 5]
        self.assertIn(3, myList)  # 断言3在myList中存在

    def test_not_in(self):
        myTuple = (1, 2, 3, 4, 5)
        self.assertNotIn(6, myTuple)  # 断言6不在myTuple中存在

5. assertRaises()
该方法用于判断一个特定的异常是否被抛出。

import unittest

def divide(x):
    return 1/x

class MyTest(unittest.TestCase):
    def test_exception(self):
        self.assertRaises(ZeroDivisionError, divide, 0)  # 断言在divide(0)的情况下会抛出ZeroDivisionError异常

以上是unittest中几个常用的断言方法的代码示例。使用这些断言方法，我们可以方便地对代码进行单元测试，确保其功能的正确性。