为什么这段代码只工作了一半？

Question

我有以下测试

def test_employees_not_arround_for_more_than_3_rounds(self):
    self.game_object.generate_workers()

    people_in_list_turn_1 = self.game_object.employees[:]
    self.game_object.next_turn()
    self.game_object.generate_workers()
    self.game_object.next_turn()
    self.game_object.generate_workers()
    self.game_object.next_turn()

    for employee in people_in_list_turn_1:
        self.assertFalse(employee in self.game_object.employees)

基本上，它会生成随机数量的工人并将其添加到我的game_object.employees 列表中。 当我调用game_object.next_turn 函数时，每个员工都有一个turns_unemployed 变量来保存他们失业的轮数，一旦达到3，该工人将完全从game_object.employees 列表中删除。

下面是game_object.py的实现代码：

def generate_workers(self):
    workersToAdd = range(random.randrange(1,8))
    for i in workersToAdd:
        self.__employees.append(Employee())

def next_turn(self):
    self.__current_turn += 1
    self.__call_employees_turn_function()
    self.__remove_workers_unemployed_for_3_rounds()

def __call_employees_turn_function(self):
    for employee in self.employees:
        employee.turn()

def __remove_workers_unemployed_for_3_rounds(self):
    for employee in self.employees:
        if employee.turns_unemployed >= 3:
            self.employees.remove(employee)

我已经有一个测试可以检查 turns_unemployed 变量在调用 employee.turn() 时实际上增加了 1，所以我知道它有效...

真正让我烦恼的是，我的测试只在我运行它的 50% 的时间里有效，我不知道为什么......有人看到任何可能导致任何差异的东西吗？

顺便说一句，运行 Python 3.2.2

Answer 1

在__remove_workers_unemployed_for_3_rounds 中迭代列表时，您正在从列表中删除项目，因此循环会跳过您希望它删除的项目。您需要遍历列表的副本。

def __remove_workers_unemployed_for_3_rounds(self):
    for employee in self.employees[:]:
        if employee.turns_unemployed >= 3:
            self.employees.remove(employee)

例子：

你每回合产生 2 名新员工。在第 4 回合，您有 2 名员工要移除（列表中的前两名）。您开始迭代并删除第一个。该列表现在只有五个项目，但迭代继续并查看第二个项目。问题是第二个项目不再是第二个员工，而是第三个。第二名员工将保留在列表中，您的测试将失败。只有在第一轮只生成一个员工时，您的测试才有效。

Answer 2

Hugo 对导致您的问题的原因可能是正确的；在迭代列表时，您无法从列表中删除项目。这是另一个可能的问题，当您创建员工时，您将他们放在一个名为 __employees 的列表中，即

def generate_workers(self):
    workersToAdd = range(random.randrange(1,8))
    for i in workersToAdd:
        self.__employees.append(Employee())

但是当您稍后对其进行迭代时，您使用的是一个名为 employees 的列表，即

def __call_employees_turn_function(self):
    for employee in self.employees:
        employee.turn()

def __remove_workers_unemployed_for_3_rounds(self):
    for employee in self.employees:
        if employee.turns_unemployed >= 3:
            self.employees.remove(employee)

但我不知道这是否与您的问题有关，因为我看不到您的其余代码 - 我什至不确定它们是否属于同一类。您可能应该发布您可以获得的最小的完整代码，但它存在问题 - 这样人们就可以实际运行您的代码并自己重现问题。

Answer 3

不要修改您正在迭代的容器。

保留副本以进行迭代也是一种丑陋的技巧，如果您必须非常精确地了解对象身份与对象相等性，它可能会在以后让您绊倒。它也只是一团糟。

有一个更简单的方法：采用函数式编程方法。使用规则“原容器中不满足被移除条件的所有内容”创建一个新容器，然后开始使用它而不是原容器。

def __remove_workers_unemployed_for_3_rounds(self):
    self.employees = filter(lambda e: e.turns_unemployed < 3, self.employees)
    # Or with a list comprehension:
    # self.employees = [e for e in self.employees if e.turns_unemployed < 3]
    # if you find that more readable.