def test_dec(self):
"""Check decorators"""
m = IMP.Model("decorators")
ps = IMP._TrivialDecorators()
for i in range(10):
ps.append(IMP._TrivialDecorator.setup_particle(IMP.Particle(m)))
pso = IMP._pass_decorators(ps)
self._equal_lists(ps, pso)
def test_dec(self):
"""Check decorators"""
m = IMP.Model("decorators")
ps = IMP._TrivialDecorators()
for i in range(10):
ps.append(IMP._TrivialDecorator.setup_particle(IMP.Particle(m)))
pso = IMP._pass_decorators(ps)
self._equal_lists(ps, pso)
def test_dec_conv(self):
"""Checking decorator conversions work"""
m = IMP.Model("decorator conversion")
p0 = IMP.Particle(m)
p1 = IMP.Particle(m)
print(p0)
print(p1)
d0 = IMP._TrivialDecorator.setup_particle(p0)
d1 = IMP._TrivialTraitsDecorator.setup_particle(p1)
print(d0)
print(d1)
d1p = IMP._TrivialDecorator.setup_particle(p1)
print(d1p)
ps = [p0, d1]
pso = IMP._pass_decorators(ps)
print(pso[0])
print(pso[1])
self.assertEqual(pso[0], d0)
self.assertEqual(pso[1], d1p)
def test_dec_conv(self):
"""Checking decorator conversions work"""
m = IMP.Model("decorator conversion")
p0 = IMP.Particle(m)
p1 = IMP.Particle(m)
print(p0)
print(p1)
d0 = IMP._TrivialDecorator.setup_particle(p0)
d1 = IMP._TrivialTraitsDecorator.setup_particle(p1)
print(d0)
print(d1)
d1p = IMP._TrivialDecorator.setup_particle(p1)
print(d1p)
ps = [p0, d1]
pso = IMP._pass_decorators(ps)
print(pso[0])
print(pso[1])
self.assertEqual(pso[0], d0)
self.assertEqual(pso[1], d1p)
