from pylearn2.datasets.csv_dataset import CSVDataset
import numpy as np

def test_loading():
    d = CSVDataset(path = 'test.csv', expect_headers = False)
    assert(np.array_equal(d.X, np.array([[1., 2., 3.], [4., 5., 6.]])))
    assert(np.array_equal(d.y, np.array([0., 1.])))
    
    


@@ -86,16 +86,101 @@
 ding():%0A
+    test_path = os.path.join(pylearn2.__path__%5B0%5D, 'datasets', 'tests', 'test.csv') %0A
     d = 
@@ -197,26 +197,25 @@
 (path = 
-'
 test
-.csv'
+_path
 , expect

test file path