给段unittest示例代码看看
时间: 2023-02-09 08:42:02 浏览: 114
当然可以,下面是一段简单的 Python unittest 示例代码:
```
import unittest
def add(x, y):
return x + y
class TestAdd(unittest.TestCase):
def test_add(self):
self.assertEqual(add(1, 2), 3)
self.assertEqual(add(-1, 1), 0)
self.assertEqual(add(-1, -1), -2)
if __name__ == '__main__':
unittest.main()
```
该代码定义了一个简单的加法函数 `add()`,并在 `TestAdd` 类中编写了三个测试用例来测试该函数。在代码末尾,我们使用 `unittest.main()` 来运行测试。
执行该代码,会打印类似于下面的输出:
```
..
----------------------------------------------------------------------
Ran 2 tests in 0.000s
OK
```
其中 ".." 表示测试用例都通过了。
相关问题
写一段unittest的代码
好的,以下是一个简单的unittest代码示例:
```python
import unittest
def add_numbers(a, b):
return a + b
class TestAddNumbers(unittest.TestCase):
def test_add_positive_numbers(self):
self.assertEqual(add_numbers(2, 3), 5)
def test_add_negative_numbers(self):
self.assertEqual(add_numbers(-2, -3), -5)
def test_add_mixed_numbers(self):
self.assertEqual(add_numbers(2, -3), -1)
self.assertEqual(add_numbers(-2, 3), 1)
if __name__ == '__main__':
unittest.main()
```
这个示例定义了一个`add_numbers`函数,用于计算两个数字的和。然后,通过继承`unittest.TestCase`类来创建一个测试类,其中定义了三个测试方法,分别测试了两个正数相加、两个负数相加以及一个正数和一个负数相加的情况。
在`main`函数中,我们调用`unittest.main()`来运行所有的测试方法,并检查它们是否通过了断言。如果在测试过程中发现了任何错误,unittest会抛出一个异常,并显示错误的详细信息。
unittest断言方法代码示例
unittest是Python中常用的单元测试框架,它提供了丰富的断言方法用于判断测试结果是否符合预期。下面是一些unittest断言方法的代码示例:
1. assertEqual()
该方法用于判断两个值是否相等。
```python
import unittest
class MyTest(unittest.TestCase):
def test_equal(self):
self.assertEqual(1+1, 2) # 断言1+1等于2
```
2. assertNotEqual()
该方法用于判断两个值是否不相等。
```python
import unittest
class MyTest(unittest.TestCase):
def test_not_equal(self):
self.assertNotEqual(1+1, 3) # 断言1+1不等于3
```
3. assertTrue()和assertFalse()
这两个方法分别用于判断表达式是否为True和False。
```python
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()
这两个方法用于判断一个值是否在列表或元组中存在或不存在。
```python
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()
该方法用于判断一个特定的异常是否被抛出。
```python
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中几个常用的断言方法的代码示例。使用这些断言方法,我们可以方便地对代码进行单元测试,确保其功能的正确性。
阅读全文