def test_trajectory_show_hide_sending_cooridnates():
view = NGLWidget()
traj0 = pt.datafiles.load_tz2()
traj1 = pt.datafiles.load_trpcage()
view.add_trajectory(nv.PyTrajTrajectory(traj0))
view.add_trajectory(nv.PyTrajTrajectory(traj1))
for traj in view._trajlist:
nt.assert_true(traj.shown)
view.frame = 1
def copy_coordinate_dict(view):
# make copy to avoid memory free
return dict((k, v.copy()) for k, v in view.coordinates_dict.items())
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide 0
view.hide([
0,
])
nt.assert_false(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
# update frame so view can update its coordinates
view.frame = 2
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
aa_eq(coordinates_dict[1], traj1[2].xyz)
# hide 0, 1
view.hide([0, 1])
nt.assert_false(view._trajlist[0].shown)
nt.assert_false(view._trajlist[1].shown)
view.frame = 3
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
nt.assert_equal(coordinates_dict[1].shape[0], 0)
# slicing, show only component 1
view[1].show()
view.frame = 0
nt.assert_false(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
aa_eq(coordinates_dict[1], traj1[0].xyz)
# show all
view[1].show()
view[0].show()
view.frame = 1
nt.assert_true(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide all
view[1].hide()
view[0].hide()
view.frame = 2
nt.assert_false(view._trajlist[0].shown)
nt.assert_false(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
nt.assert_equal(coordinates_dict[1].shape[0], 0)
def test_trajectory_show_hide_sending_cooridnates():
view = NGLWidget()
traj0 = pt.datafiles.load_tz2()
traj1 = pt.datafiles.load_trpcage()
view.add_trajectory(nv.PyTrajTrajectory(traj0))
view.add_trajectory(nv.PyTrajTrajectory(traj1))
for traj in view._trajlist:
nt.assert_true(traj.shown)
view.frame = 1
def copy_coordinate_dict(view):
# make copy to avoid memory free
return dict((k, v.copy()) for k, v in view.coordinates_dict.items())
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide 0
view.hide([0,])
nt.assert_false(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
# update frame so view can update its coordinates
view.frame = 2
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
aa_eq(coordinates_dict[1], traj1[2].xyz)
# hide 0, 1
view.hide([0, 1])
nt.assert_false(view._trajlist[0].shown)
nt.assert_false(view._trajlist[1].shown)
view.frame = 3
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
nt.assert_equal(coordinates_dict[1].shape[0], 0)
# slicing, show only component 1
view[1].show()
view.frame = 0
nt.assert_false(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
aa_eq(coordinates_dict[1], traj1[0].xyz)
# show all
view[1].show()
view[0].show()
view.frame = 1
nt.assert_true(view._trajlist[0].shown)
nt.assert_true(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide all
view[1].hide()
view[0].hide()
view.frame = 2
nt.assert_false(view._trajlist[0].shown)
nt.assert_false(view._trajlist[1].shown)
coordinates_dict = copy_coordinate_dict(view)
nt.assert_equal(coordinates_dict[0].shape[0], 0)
nt.assert_equal(coordinates_dict[1].shape[0], 0)
def test_trajectory_show_hide_sending_cooridnates():
view = NGLWidget()
traj0 = pt.datafiles.load_tz2()
traj1 = pt.datafiles.load_trpcage()
view.add_trajectory(nv.PyTrajTrajectory(traj0))
view.add_trajectory(nv.PyTrajTrajectory(traj1))
for traj in view._trajlist:
assert traj.shown
view.frame = 1
def copy_coordinate_dict(view):
# make copy to avoid memory free
return {k: v.copy() for k, v in view._coordinates_dict.items()}
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide 0
view.hide([
0,
])
assert not view._trajlist[0].shown
assert view._trajlist[1].shown
# update frame so view can update its coordinates
view.frame = 2
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
aa_eq(coordinates_dict[1], traj1[2].xyz)
# hide 0, 1
view.hide([0, 1])
assert not view._trajlist[0].shown
assert not view._trajlist[1].shown
view.frame = 3
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
assert coordinates_dict[1].shape[0] == 0
# slicing, show only component 1
view[1].show()
view.frame = 0
assert not view._trajlist[0].shown
assert view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
aa_eq(coordinates_dict[1], traj1[0].xyz)
# show all
view[1].show()
view[0].show()
view.show(indices='all')
view.show(indices=[
0,
])
view.show(indices=[0, 1])
view.frame = 1
assert view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide all
view[1].hide()
view[0].hide()
view.frame = 2
assert not view._trajlist[0].shown
assert not view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
assert coordinates_dict[1].shape[0] == 0
def test_trajectory_show_hide_sending_cooridnates():
view = NGLWidget()
traj0 = pt.datafiles.load_tz2()
traj1 = pt.datafiles.load_trpcage()
view.add_trajectory(nv.PyTrajTrajectory(traj0))
view.add_trajectory(nv.PyTrajTrajectory(traj1))
for traj in view._trajlist:
assert traj.shown
view.frame = 1
def copy_coordinate_dict(view):
# make copy to avoid memory free
return dict((k, v.copy()) for k, v in view.coordinates_dict.items())
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide 0
view.hide([0,])
assert not view._trajlist[0].shown
assert view._trajlist[1].shown
# update frame so view can update its coordinates
view.frame = 2
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
aa_eq(coordinates_dict[1], traj1[2].xyz)
# hide 0, 1
view.hide([0, 1])
assert not view._trajlist[0].shown
assert not view._trajlist[1].shown
view.frame = 3
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
assert coordinates_dict[1].shape[0] == 0
# slicing, show only component 1
view[1].show()
view.frame = 0
assert not view._trajlist[0].shown
assert view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
aa_eq(coordinates_dict[1], traj1[0].xyz)
# show all
view[1].show()
view[0].show()
view.show(indices='all')
view.show(indices=[0,])
view.show(indices=[0, 1])
view.frame = 1
assert view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
aa_eq(coordinates_dict[0], traj0[1].xyz)
aa_eq(coordinates_dict[1], traj1[1].xyz)
# hide all
view[1].hide()
view[0].hide()
view.frame = 2
assert not view._trajlist[0].shown
assert not view._trajlist[1].shown
coordinates_dict = copy_coordinate_dict(view)
assert coordinates_dict[0].shape[0] == 0
assert coordinates_dict[1].shape[0] == 0
