class WorldTest(unittest.TestCase): def setUp(self): self.world = World() def testRange(self): p = LinearParticle(1, 0) q = LinearParticle(3, 0) self.world.bind(p) self.world.bind(q) self.assert_(self.world.num_particles() == 2) v = self.world.particles_in_range(p, 1) self.assert_(len(v) == 0) v = self.world.particles_in_range(p, 2) self.assert_(len(v) == 1) self.assert_(v[0].this == q.this) # same underlying object hopefully def testObstacles(self): ls = Obstacle(Vec2d(0, 0), Vec2d(1, 2)) self.world.bind(ls) self.assert_(len(self.world.get_obstacles()))