use msmb_data

Author: Carlos Hernandez

osprey/tests/test_dataset_loader.py

@@ -147,11 +147,13 @@ def test_DSVDatasetLoader_1():
 
 
 def test_MDTrajDatasetLoader_1():
-    from mdtraj.testing import get_fn
-
-    loader = MDTrajDatasetLoader(get_fn('legacy_msmbuilder_trj0.lh5'))
+    from msmbuilder.example_datasets import FsPeptide
+    fs_pept = FsPeptide()
+    
+    loader = MDTrajDatasetLoader(os.path.join(fs_pept.data_dir, '*.xtc'),
+                                 topology=os.path.join(fs_pept.data_dir, 'fs-peptide.pdb'))
     X, y = loader.load()
-    assert X[0].n_frames == 501
+    assert len(X) == 28
     assert y is None