def setUp(self):
magnet1 = MagnetModel(1.0, 0.0, 1.5)
magnet2 = MagnetModel(-1.0, 1.0, 1.5)
magnet3 = MagnetModel(-1.0, -1.0, 1.5)
magnet4 = MagnetModel(0.0, 0.0, 0.5)
self.magnets = [magnet1, magnet2, magnet3, magnet4]
self.plane_distance = 0.25
self.friction = 0.3
self.gravity_pullback = 0.5
self.pendulum = PendulumModel(self.friction, self.gravity_pullback,
self.plane_distance)
self.pendulum.magnets = self.magnets
self.time_step = 0.01
self.integrator = EulerIntegrator(self.time_step)
self.r = 255
self.g = 0
self.b = 0
self.image_generator = BasicImageGenerator(self.r, self.g, self.b)
self.size = 5
self.resolution = 640
self.basins_generator = BasinsGenerator(self.size, self.resolution)
self.basins_generator.pendulum_model = self.pendulum
self.basins_generator.integrator = self.integrator
self.basins_generator.image_generator = self.image_generator
