def test_simple(self): """Check overloading of particle methods works""" m = IMP.Model() p = IMP.Particle(m) IMP._overloaded_particles(m, [p.get_index()]) IMP._overloaded_particles(p) IMP._overloaded_particles([p]) td = IMP._TrivialDecorator.setup_particle(p) IMP._overloaded_particles([td]) IMP._overloaded_particles(IMP._TrivialDecorator(p)) IMP._implicit_particles(IMP._ImplicitParticles(m, [p.get_index()])) IMP._implicit_particles(p) IMP._implicit_particles([p]) IMP._implicit_particles([td]) IMP._implicit_particles(IMP._TrivialDecorator(p))
def test_simple(self): """Check overloading of particle methods works""" m = IMP.Model() p = IMP.Particle(m) IMP._overloaded_particles(m, [p.get_index()]) IMP._overloaded_particles(p) IMP._overloaded_particles([p]) td = IMP._TrivialDecorator.setup_particle(p) IMP._overloaded_particles([td]) IMP._overloaded_particles(IMP._TrivialDecorator(p)) IMP._implicit_particles( IMP._ImplicitParticles(m, [p.get_index()])) IMP._implicit_particles(p) IMP._implicit_particles([p]) IMP._implicit_particles([td]) IMP._implicit_particles(IMP._TrivialDecorator(p))
def test_comparisons(self): """Test comparisons of particles and decorators""" (model, particles) = self.setup() p0a = particles[0] p0b = model.get_particle(model.get_particle_indexes()[0]) self.assertEqual(p0a, p0b) td0a = IMP._TrivialDecorator.setup_particle(p0a) td0b = IMP._TrivialDecorator(p0b) self.assertEqual(td0a, td0b) self.assertEqual(td0a, p0a)