def test_forget_direction_waiting(): for value in 1, 2, 3, 4, -2, -12, -3: b = BeeClust(loner(value), p_changedir=0) lo = loner(-1) b.map[0, 0] = lo[0, 0] = 5 b.map[-1, -1] = lo[-1, -1] = 6 b.map[0, -1] = lo[0, -1] = 7 b.forget() assert (b.map == lo).all()
def test_forget_is_callable(): b = BeeClust(zeros8((2, 2))) b.forget()