def test_modify_frame_use_Pipeline(self):
traj = pt.iterload(fn("tz2.ortho.nc"), fn("tz2.ortho.parm7"))
dslist = CpptrajDatasetList()
dslist.add('topology', name='mytop')
# add a new topology
dslist[0].data = pt.strip(traj.top, ':WAT')
commands = [
'autoimage',
'strip :WAT',
'createcrd mycrd',
]
actlist = Pipeline(commands, top=traj.top, dslist=dslist)
for frame in traj:
actlist.compute(frame)
aa_eq(dslist['mycrd'].xyz,
pt.get_coordinates(traj, mask='!:WAT', autoimage=True))
def test_reference(self):
traj = pt.iterload(fn("tz2.ortho.nc"), fn("tz2.ortho.parm7"))
# store reference
dslist = CpptrajDatasetList()
ref = dslist.add('reference')
ref.top = traj.top
ref.append(traj[3])
fi = pt.pipe(traj, ['autoimage', 'rms refindex 0 @CA'], dslist=dslist)
xyz = np.array([frame.xyz.copy() for frame in fi])
t0 = (traj[:].autoimage().superpose(ref=traj[3], mask='@CA'))
aa_eq(xyz, t0.xyz)
t1 = traj[:].autoimage()
aa_eq(pt.rmsd(t1, ref=traj[3], mask='@CA'), dslist[-1].values)
def test_constructor_from_command_list_TrajectoryIterator(self):
traj = pt.iterload("./data/Tc5b.x", "./data/Tc5b.top")
commands = [
'rmsd @CA', 'distance :3 :7', 'distance :3 :7', 'vector :2 :3'
]
dslist = CpptrajDatasetList()
actlist = ActionList(commands, traj.top, dslist=dslist)
d0 = dslist.add('ref_frame', 'my_ref')
d0.add_frame(traj[3])
for frame in traj:
actlist.compute(frame)
aa_eq(pt.rmsd(traj, mask='@CA'), dslist[0])
aa_eq(pt.distance(traj, ':3 :7'), dslist[1])
aa_eq(pt.distance(traj, ':3 :7'), dslist[2])
aa_eq(pt.vector.vector_mask(traj(rmsfit=(0, '@CA')), ':2 :3'),
dslist[3].values)
def test_constructor_from_command_list_TrajectoryIterator(self):
traj = pt.iterload("./data/Tc5b.x", "./data/Tc5b.top")
commands = ['rmsd @CA', 'distance :3 :7', 'distance :3 :7',
'vector :2 :3']
dslist = CpptrajDatasetList()
actlist = ActionList(commands, traj.top, dslist=dslist)
d0 = dslist.add('ref_frame', 'my_ref')
d0.add_frame(traj[3])
for frame in traj:
actlist.compute(frame)
aa_eq(pt.rmsd(traj, mask='@CA'), dslist[0])
aa_eq(pt.distance(traj, ':3 :7'), dslist[1])
aa_eq(pt.distance(traj, ':3 :7'), dslist[2])
aa_eq(
pt.vector.vector_mask(
traj(rmsfit=(0, '@CA')),
':2 :3'),
dslist[3].values)
