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