def save_discrete_trajectory(filename, dtraj):
r"""Write discrete trajectory to 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.
dtraj : array-like of int
Discrete state trajectory
See also
--------
load_discrete_trajectory
Notes
-----
The discrete trajectory is stored as ndarray of integers
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])
"""
trajectory.save_discrete_trajectory(filename, dtraj)
def save_discrete_trajectory(filename, dtraj):
r"""Write discrete trajectory to 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.
dtraj : array-like of int
Discrete state trajectory
See also
--------
load_discrete_trajectory
Notes
-----
The discrete trajectory is stored as ndarray of integers
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])
"""
trajectory.save_discrete_trajectory(filename, dtraj)
def test_save_discrete_trajectory(self):
dt.save_discrete_trajectory(self.filename, self.dtraj)
dtraj_n = np.load(self.filename)
self.assertTrue(np.all(dtraj_n == self.dtraj))
def test_save_discrete_trajectory(self):
dt.save_discrete_trajectory(self.filename, self.dtraj)
dtraj_n=np.load(self.filename)
self.assertTrue(np.all(dtraj_n==self.dtraj))