def load_discrete_trajectory(filename): r"""Read discrete trajectory form binary file. Parameters ---------- filename : str The filename of the discrete state trajectory file. The filename can either contain the full or the relative path to the file. Returns ------- dtraj : (M,) ndarray of int Discrete state trajectory See also -------- save_discrete_trajectory Notes ----- The binary file is a one dimensional numpy array of integers stored in numpy .npy format. Examples -------- >>> from tempfile import NamedTemporaryFile >>> from pyemma.msm.io import load_discrete_trajectory, save_discrete_trajectory Use temporary file >>> tmpfile = NamedTemporaryFile() Discrete trajectory >>> dtraj = np.array([0, 1, 0, 0, 1, 1, 0]) Write to disk (as ascii file) >>> save_discrete_trajectory(tmpfile.name, dtraj) Read from disk >>> X = load_discrete_trajectory(tmpfile.name) >>> X array([0, 1, 0, 0, 1, 1, 0]) """ return trajectory.load_discrete_trajectory(filename)
def test_load_discrete_trajectory(self): dtraj_n = np.load(self.filename) dtraj = dt.load_discrete_trajectory(self.filename) self.assertTrue(np.all(dtraj_n == dtraj))
def test_load_discrete_trajectory(self): dtraj_n=np.load(self.filename) dtraj=dt.load_discrete_trajectory(self.filename) self.assertTrue(np.all(dtraj_n==dtraj))