python断言模块的封装
时间: 2023-10-01 09:06:03 浏览: 81
断言(assert)是Python中的一个重要的调试工具,可以用来检查程序中的某些条件是否为真。Python中的断言模块(assert module)提供了一些函数,可以用来进行断言检查,包括:
- assertEqual(a, b):断言a等于b
- assertNotEqual(a, b):断言a不等于b
- assertTrue(a):断言a为真
- assertFalse(a):断言a为假
- assertIn(a, b):断言a在b中
- assertNotIn(a, b):断言a不在b中
为了方便使用,我们可以对这些函数进行封装,例如:
```python
import unittest
class Assert(unittest.TestCase):
def assertEqual(self, a, b):
super().assertEqual(a, b)
def assertNotEqual(self, a, b):
super().assertNotEqual(a, b)
def assertTrue(self, a):
super().assertTrue(a)
def assertFalse(self, a):
super().assertFalse(a)
def assertIn(self, a, b):
super().assertIn(a, b)
def assertNotIn(self, a, b):
super().assertNotIn(a, b)
```
这样,我们就可以在测试代码中使用这些封装后的函数进行断言检查了,例如:
```python
class TestAssert(Assert):
def test_assert(self):
self.assertEqual(1, 1)
self.assertNotEqual(1, 2)
self.assertTrue(True)
self.assertFalse(False)
self.assertIn(1, [1, 2, 3])
self.assertNotIn(4, [1, 2, 3])
```
这个例子中,我们继承了unittest.TestCase类,并将assert模块中的函数封装到了Assert类中。在测试代码中,我们使用了这些封装后的函数进行断言检查。
阅读全文