python - 成员资格测试结果与 list 和 csv.reader 不同

Question

如果我使用列表检查成员资格与使用 csv.reader 对象检查成员资格，我会得到不同的结果。 下面使用 unittest 模块。

成员(member)资格的 csv.reader 测试

with open("file.tab", 'rb') as f:
    reader = csv.reader(f, delimiter='\t')
    self.assertTrue(['1', '2', '3', '4'] in reader)
    self.assertTrue(['2', '3', '4', '5'] in reader)
    self.assertTrue(['3', '4', '5', '6'] in reader)

成员列表测试

with open("file.tab", 'rb') as f:
    reader = csv.reader(f, delimiter='\t')
    reader = [record for record in reader]
    self.assertTrue(['1', '2', '3', '4'] in reader)
    self.assertTrue(['2', '3', '4', '5'] in reader)
    self.assertTrue(['3', '4', '5', '6'] in reader)

我知道 file.tab 包含我正在测试的三个记录的条目，但第三个断言在使用 csv.reader 时出现“False is not true”并在使用列表时通过。

csv.reader 是一个生成器； docs没有明确说，但因为我可以用尽它，我认为这意味着它是一个发电机。我的想法是这可能是原因，但以下测试只打印 true:

x = xrange(5)
for m in range(5):
    for n in range(5):
        print m in x
        print n in x

这让我觉得使用生成器测试成员资格没有问题。

为什么当我使用 csv.reader 时第三个 assert 语句的计算结果与我使用列表时不同？

Answer 1

你运气不好 -- xrange 实际上不是一个生成器，而是它自己的一种特殊类型，它的行为是惰性的，因此可以让你误以为它是一个生成器。

>>> x = xrange(10)
>>> 5 in x
True
>>> 5 in x
True

但是

>>> it = iter(range(10))
>>> 5 in it
True
>>> 5 in it
False

所以您的逻辑是正确的:reader 实例可以用完，但列表不能，这就是为什么成员测试可以根据内容返回不同答案的原因。请注意，虽然成员资格测试可能会短路，因此在出现积极结果的情况下它们不必耗尽:

>>> it = iter(range(10))
>>> 3 in it
True
>>> next(it)
4