def test_update_best(self):
"""Test update best result."""
env = Environment(config={})
old_best = Information(position=None, fitness=5)
better_best = Information(position=None, fitness=3)
worse_best = Information(position=None, fitness=7)
particle = Particle(environment=env, current=old_best)
assert particle.best == old_best
particle.current = better_best
assert particle.best == better_best
particle.current = worse_best
assert particle.best == better_best
def test_register_new_local_best_result(self):
"""Register new local best result."""
class Environment(object):
def setInfo(self, info):
self.info = info
def updateParticleBest(self, info):
assert info == self.info
def register(self, particle):
assert self == particle.environment
env = Environment()
position = matrix([1, 2, 3])
fitness = 5
info = Information(position=position, fitness=fitness)
env.setInfo(Information(position=position, fitness=fitness))
particle = Particle(environment=env, best=info)
position = matrix([4, 5, 6])
fitness = 4
info = Information(position=position, fitness=fitness)
env.setInfo(info)
particle.current = info
def test_register_new_local_best_result(self):
"""Register new local best result."""
class Environment(object):
def setInfo(self, info):
self.info = info
def updateParticleBest(self, info):
assert info == self.info
def register(self, particle):
assert self == particle.environment
env = Environment()
position = matrix([1, 2, 3])
fitness = 5
info = Information(position=position, fitness=fitness)
env.setInfo(Information(position=position, fitness=fitness))
particle = Particle(
environment=env, best=info)
position = matrix([4, 5, 6])
fitness = 4
info = Information(position=position, fitness=fitness)
env.setInfo(info)
particle.current = info
def test_update_best(self):
"""Test update best result."""
env = Environment(config={})
old_best = Information(position=None, fitness=5)
better_best = Information(position=None, fitness=3)
worse_best = Information(position=None, fitness=7)
particle = Particle(
environment=env, current=old_best)
assert particle.best == old_best
particle.current = better_best
assert particle.best == better_best
particle.current = worse_best
assert particle.best == better_best