三角形问题代码的单元测试
时间: 2024-02-13 20:05:50 浏览: 25
好的,我可以帮您编写三角形问题代码的单元测试。以下是一个Python代码示例:
```python
import unittest
from triangle import get_triangle_type
class TestTriangle(unittest.TestCase):
def test_equilateral_triangle(self):
self.assertEqual(get_triangle_type(5, 5, 5), "Equilateral")
def test_isosceles_triangle(self):
self.assertEqual(get_triangle_type(3, 3, 4), "Isosceles")
def test_scalene_triangle(self):
self.assertEqual(get_triangle_type(3, 4, 5), "Scalene")
def test_invalid_triangle(self):
self.assertEqual(get_triangle_type(1, 2, 3), "Invalid")
if __name__ == '__main__':
unittest.main()
```
在这里,我们使用Python内置的unittest模块编写了一个名为TestTriangle的测试类。其中,我们定义了四个测试方法来测试三角形函数的不同输入和输出。
我们可以使用assertEqual()方法来比较函数的实际输出和预期输出。在这里,我们期望函数返回Equilateral、Isosceles、Scalene和Invalid这四种类型的三角形。
最后,我们在文件的末尾使用unittest.main()方法来运行测试用例。
您可以根据您的实际需求修改这个示例代码。希望这可以帮到您!