python Unittest essertionError：0！= []

Question

我是Python Unittest的新手，我正在尝试访问此列表：

    def setUp(self):
        self.customers = [
            {"name": "Mary", "pets": [], "cash": 1000},
            {"name": "Alan", "pets": [], "cash": 50},
            {"name": "Richard", "pets": [], "cash": 100},
        ]

要进行此测试：

    def test_customer_pet_count(self):
        count = get_customer_pet_count(self.customers[0])
        self.assertEqual(0, count)

我创建了此功能：

def get_customer_pet_count(customer_number):
    if ["pets"] == 0 or ["pets"] == []:
        return 0
    return customer_number["pets"]

但是我一直遇到此错误： essertionError：0！= []

有人可以帮助解释我在功能中做错了什么吗？

Answer 1

让我们看一下这部分，get_customer_pet_count函数：

def get_customer_pet_count(customer_number):
    if ["pets"] == 0 or ["pets"] == []:
        return 0
    return customer_number["pets"]

首先，您不是将其传递给“客户号码”或索引，而是将其传递给实际的客户字典。像{“名称”：“玛丽”，“宠物”：[]，“现金”：1000}。

其次，此比较：[“ PETS”] == 0检查“如果一个带有一个元素的列表，即字符串'Pets'等于数字0”。这永远不可能是真的。 永远不会 等于 一个数字。

列表 使用一个元素，字符串“宠物”等于一个空列表”。那也永远不会是真的。空列表不能等于非空的列表。

如果您将其编写为def get_customer_pet_count（customer）：，则可能更清楚。您正在将其传递给客户 info 的字典，而不是客户 number 。另外，您的功能说 pet_count ，因此应该是宠物列表的长度：

def get_customer_pet_count(customer):
    return len(customer["pets"])

_{*忽略用户定义的类型伪造该行为。}

Answer 2

self.customers [0]是空列表[]。您应该使用len（self.customers [0]）或这样修改函数：

def get_customer_pet_count(customer_number):
    if customer_number == 0 or customer_number == []:
        return 0
    return customer_number