def setUp(self):
self.max_velocity = [10.0, 10.0]
self.bounds = [[0, 10], [0, 10]]
self.best = PSOParticle(score=0.0,
position=[1.0, 1.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds)
self.particle = PSOParticle(score=0.1,
position=[3.0, 4.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds)
def setUp(self):
self.max_velocity = [10.0, 10.0]
self.bounds = [[0, 10], [0, 10]]
config = {
"max_population": 5,
"max_velocity": self.max_velocity,
"bounds": self.bounds,
"objective_function": self.obj_func
}
self.particle = PSOParticle(score=0.0,
position=[1.0, 1.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds)
self.generator = PSOParticleGenerator(config)
def setUp(self):
self.max_velocity = [10.0, 10.0]
self.bounds = [[0, 10], [0, 10]]
self.best = PSOParticle(
score=0.0,
position=[1.0, 1.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds
)
self.particle = PSOParticle(
score=0.1,
position=[3.0, 4.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds
)
class PSOParticleTests(unittest.TestCase):
def setUp(self):
self.max_velocity = [10.0, 10.0]
self.bounds = [[0, 10], [0, 10]]
self.best = PSOParticle(score=0.0,
position=[1.0, 1.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds)
self.particle = PSOParticle(score=0.1,
position=[3.0, 4.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds)
def tearDown(self):
pass
def test_update_velocity(self):
# update velocity
vel_before = list(self.particle.velocity)
self.particle.update_velocity(self.best, 2.0, 2.0)
vel_after = list(self.particle.velocity)
# print "VELOCITY BEFORE", vel_before
# print "VELOCITY AFTER", vel_after
# assert
self.assertNotEquals(vel_before, vel_after)
# update velocity with max velocity
self.particle.velocity = [100, 100]
vel_before = list(self.particle.velocity)
self.particle.update_velocity(self.best, 2.0, 2.0)
vel_after = list(self.particle.velocity)
# print "VELOCITY BEFORE", vel_before
# print "VELOCITY AFTER", vel_after
# assert
self.assertNotEquals(vel_before, vel_after)
self.assertEquals(vel_after, self.max_velocity)
def test_check_over_bounds(self):
# check no boundary violation
self.particle.position = [5, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# assert
self.assertEquals(before, after)
# check x-axis under bounds
self.particle.position = [-1, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "x-axis under bounds[BEFORE]:", before
# print "x-axis under bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [1, 5])
# check x-axis over bounds
self.particle.position = [11, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "x-axis over bounds[BEFORE]:", before
# print "x-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [9, 5])
# check y-axis under bounds
self.particle.position = [5, -1]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "y-axis over bounds[BEFORE]:", before
# print "y-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [5, 1])
# check y-axis over bounds
self.particle.position = [5, 11]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "y-axis over bounds[BEFORE]:", before
# print "y-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [5, 9])
def test_update_position(self):
# update position
pos_before = list(self.particle.position)
self.particle.update_position()
pos_after = list(self.particle.position)
# print "VELOCITY", self.particle.velocity
# print "POSITION BEFORE", pos_before
# print "POSITION AFTER", pos_after
# assert
self.assertNotEquals(pos_before, pos_after)
class PSOParticleTests(unittest.TestCase):
def setUp(self):
self.max_velocity = [10.0, 10.0]
self.bounds = [[0, 10], [0, 10]]
self.best = PSOParticle(
score=0.0,
position=[1.0, 1.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds
)
self.particle = PSOParticle(
score=0.1,
position=[3.0, 4.0],
velocity=[1.0, 1.0],
max_velocity=self.max_velocity,
bounds=self.bounds
)
def tearDown(self):
pass
def test_update_velocity(self):
# update velocity
vel_before = list(self.particle.velocity)
self.particle.update_velocity(self.best, 2.0, 2.0)
vel_after = list(self.particle.velocity)
# print "VELOCITY BEFORE", vel_before
# print "VELOCITY AFTER", vel_after
# assert
self.assertNotEquals(vel_before, vel_after)
# update velocity with max velocity
self.particle.velocity = [100, 100]
vel_before = list(self.particle.velocity)
self.particle.update_velocity(self.best, 2.0, 2.0)
vel_after = list(self.particle.velocity)
# print "VELOCITY BEFORE", vel_before
# print "VELOCITY AFTER", vel_after
# assert
self.assertNotEquals(vel_before, vel_after)
self.assertEquals(vel_after, self.max_velocity)
def test_check_over_bounds(self):
# check no boundary violation
self.particle.position = [5, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# assert
self.assertEquals(before, after)
# check x-axis under bounds
self.particle.position = [-1, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "x-axis under bounds[BEFORE]:", before
# print "x-axis under bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [1, 5])
# check x-axis over bounds
self.particle.position = [11, 5]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "x-axis over bounds[BEFORE]:", before
# print "x-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [9, 5])
# check y-axis under bounds
self.particle.position = [5, -1]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "y-axis over bounds[BEFORE]:", before
# print "y-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [5, 1])
# check y-axis over bounds
self.particle.position = [5, 11]
before = list(self.particle.position)
self.particle.check_over_bounds()
after = list(self.particle.position)
# print "y-axis over bounds[BEFORE]:", before
# print "y-axis over bounds[AFTER]:", after
# assert
self.assertEquals(self.particle.position, [5, 9])
def test_update_position(self):
# update position
pos_before = list(self.particle.position)
self.particle.update_position()
pos_after = list(self.particle.position)
# print "VELOCITY", self.particle.velocity
# print "POSITION BEFORE", pos_before
# print "POSITION AFTER", pos_after
# assert
self.assertNotEquals(pos_before, pos_after)
