def test_dec_to_part(self): """Checking conversion of decorators to particles""" m = IMP.Model("particle decorator conversion") ps = IMP._TrivialDecorators() rps = [] for i in range(10): ps.append(IMP._TrivialDecorator.setup_particle(IMP.Particle(m))) rps.append(ps[-1].get_particle()) pso = IMP._pass_particles(ps) self._equal_lists(rps, pso)
def test_no_model(self): """Check particle typemaps""" m = IMP.Model("particle typemaps") ps = [] for i in range(0, 10): ps.append(IMP.Particle(m)) num = IMP._take_particles(ps) self.assertEqual(num, len(ps)) num = IMP._take_particles(m, ps) self.assertEqual(num, len(ps)) num = IMP._take_particles(m, ps, io.BytesIO()) self.assertEqual(num, len(ps)) pps = IMP._pass_particles(ps) for i in range(len(ps)): self.assertEqual(pps[i], ps[i]) pso = IMP._give_particles(m) self.assertEqual(len(pso), 10) print(pso[0]) di = {} for p in ps: di[p] = IMP._TrivialDecorator.setup_particle(p) rps = IMP._take_particles(list(di.values()))