python - 返回错误类型，如何修复？

Question

我有一个错误的问题，因为我的代码中给出了错误的类型。

问题概要:

我试图回答的第一个问题写在这里:

哪两个部门最常相互混淆以及有多少次混淆？

0) 非必需消费品
1) 必需消费品
2) 能源
3)财务状况 4) 医疗保健
5) 工业
6) 信息技术
7) Material
8) 房地产
9) 电信服务
10) 实用工具

我尝试过的代码是这样的:

def question_1():
    ans1 = np.array([5, 7])
    num_scen1 = 5
    return ans1, num_scen1

我的代码下面的断言语句如下所示:

ans1, num_scen1 = question_1()
assert type(ans1) == list, "Incorrect type"
assert type(ans1[0]) == tuple, "Incorrect type"
assert type(ans1[0][0]) == int, "Incorrect type"
assert type(num_scen1) == int, "Incorrect type"

我的代码触发了此错误:

如何通过使用正确的类型来帮助修复我的代码以传递断言语句？

Answer 1

1. 函数question_1()返回第一个元素作为numpy数组类型而不是列表，所以我们应该将它与np.ndarray.

2. 由于我们在这里检查变量的type，我们可能更喜欢isinstance()方法，因此更新代码如下:

   ans1, num_scen1 = question_1()
   assert isinstance(ans1, np.ndarray), "Incorrect Type"
   assert isinstance(ans1[0], tuple), "Incorrect type"
   assert isinstance(ans1[0][0], int), "Incorrect type"
   assert isinstance(num_scen1, int), "Incorrect type"