python写一个接口,入参为一个数组,再给一个测试实例
时间: 2024-05-07 12:15:53 浏览: 35
以下是一个简单的Python接口,接受一个数组作为输入,返回数组的最大值和最小值:
```python
from typing import List
def array_stats(arr: List[int]) -> dict:
if not arr:
return {"max": None, "min": None}
max_val = max(arr)
min_val = min(arr)
return {"max": max_val, "min": min_val}
```
这个接口使用了Python的类型注解,来指定输入参数和返回值的类型。输入参数是一个整数类型的数组(List[int]),返回值是一个字典类型,包含最大值和最小值两个键。
以下是一个简单的测试实例,使用了Python的unittest模块:
```python
import unittest
class TestArrayStats(unittest.TestCase):
def test_array_stats(self):
arr = [1, 2, 3, 4, 5]
expected_output = {"max": 5, "min": 1}
self.assertEqual(array_stats(arr), expected_output)
if __name__ == '__main__':
unittest.main()
```
这个测试实例创建了一个包含整数1到5的数组,并期望接口返回最大值为5,最小值为1。使用unittest模块的assertEqual方法来比较实际输出和期望输出。如果测试通过,则不会有任何输出。如果测试失败,则会打印错误信息。
阅读全文