如何测试一个函数只接受某种类型？答案

【问题标题】：How to test that a function only accepts a certain type?如何测试一个函数只接受某种类型？
【发布时间】：2016-09-04 13:12:30
【问题描述】：

我想确保twist_data 只获取整数列表作为参数，并添加单元测试。

 def twist_data(n):
     n=list(n)
     a=[]
     b=[]
     for i in n:
         if i<=0:
             a.append(i)
         if i>0:
             b.append(i)
     n=[len(a),sum(b)]
     return n


 gum=[2,4,5,-6,-7,]
 print (twist_data (gum))

【问题讨论】：

你的问题是如何确保函数只有在n 是一个列表时才能完成？
我的意思是。我希望能够以 pyhonic 方式测试参数是整数列表。因此，如果有人为 dict 或 tuple 调用该函数，它将给他一个自定义的错误消息。如果他调用它的元素是字符串的列表，那么它也会给他一个自定义的错误消息
我不认为有一个 Python 的方法来测试你在问什么，因为只使用列表（当任何可迭代对象都可以）不是 Python 的。同样，如果n 中有一个无法与整数比较的值（如果您使用的是 Python 3，它将包括字符串，其中像 <= 这样的比较运算符在任意之间不起作用类型）。
那我该怎么办。我试过这个：
澄清标题，根据澄清 OP 中的 cmets 改写问题

标签： python unit-testing

【解决方案1】：

def all_ints(numbers):
    return all(isinstance(number, int) for number in numbers)


def validate_list(numbers):
    return all((isinstance(numbers, list), all_ints(numbers))


def twist_data(numbers):
    if not validate_list(numbers):
        raise TypeError('"numbers" must be of type "list", where all elements integers')
    positives, negatives = [], []
    for number in numbers:
        if number <= 0:
            negatives.append(number)
        if i > 0:
            positives.append(number)
    return [len(negatives), sum(positives)]

虽然通常 Pythonic 不会进行显式类型检查，但有时这是不可避免的。这是一种非常直接的方法。然而，请求原谅比请求许可更容易，所以这是另一种可能性：

def twist_data(numbers):
    positives, negatives = [], []
    try:
        for number in numbers:
            if number <= 0:
                negatives.append(number)
            if i > 0:
                positives.append(number)
        return [len(negatives), sum(positives)]
    except TypeError:
        raise TypeError('"numbers" must be of type "list", where all elements integers')

如果您尝试在不可比较的类型之间进行比较，则会引发TypeError。然后，您可以捕获异常并使用您自己的消息重新引发。

还要尽量避免使用单字母变量名。跟踪正在发生的事情会容易得多。

【讨论】：