__author__ = 'noe'

def load_2well_discrete():
    from .double_well_discrete import DoubleWell_Discrete_Data
    return DoubleWell_Discrete_Data()

def get_bpti_test_data():
    """ Returns a dictionary containing C-alpha coordinates of a truncated
    BTBI trajectory.

    Notes
    -----
    You will have to load the data from disc yourself. See eg.
    :py:func:`pyemma.coordinates.load`.

    Returns
    -------
    res : {trajs : list, top: str}
        trajs is a list of filenames
        top is str pointing to the path of the topology file.
    """
    import os
    from glob import glob
    import pkg_resources
    path = pkg_resources.resource_filename('pyemma.coordinates.tests', 'data/')
    top = pkg_resources.resource_filename('pyemma.coordinates.tests', 'data/bpti_ca.pdb')
    trajs = sorted(filter(lambda x: x.find('mini') == -1, glob(path + os.sep + "*.xtc")))
    return {'trajs': trajs, 'top': top}

@@ -1,8 +1,65 @@
+from __future__ import (print_function, absolute_import)%0A
 __author
@@ -847,16 +847,17 @@
 a.pdb')%0A
+%0A
     traj
@@ -864,15 +864,50 @@
 s = 
-sorted(
+glob(path + os.sep + %22*.xtc%22)%0A    trajs = 
 filt
@@ -920,64 +920,104 @@
 bda 
-x: x.find('mini') == -1, glob(path + os.sep + %22*.xtc%22)))
+f: not f.endswith(%22bpti_mini.xtc%22), trajs)%0A    trajs = sorted(trajs)%0A%0A    assert len(trajs) == 3
 %0A   

fix bpti test data