Moved data for tests.

Author: MarcCote

nibabel/streamlines/tests/test_streamlines.py

@@ -9,6 +9,7 @@
 from nibabel.externals.six import BytesIO
 from nibabel.tmpdirs import InTemporaryDirectory
 
+from nibabel.testing import data_path
 from nibabel.testing import clear_and_catch_warnings
 from nose.tools import assert_equal, assert_raises, assert_true, assert_false
 
@@ -17,18 +18,16 @@
 from ..tractogram_file import TractogramFile, ExtensionWarning
 from .. import trk
 
-DATA_PATH = pjoin(os.path.dirname(__file__), 'data')
-
 DATA = {}
 
 
 def setup():
     global DATA
-    DATA['empty_filenames'] = [pjoin(DATA_PATH, "empty" + ext)
+    DATA['empty_filenames'] = [pjoin(data_path, "empty" + ext)
                                for ext in nib.streamlines.FORMATS.keys()]
-    DATA['simple_filenames'] = [pjoin(DATA_PATH, "simple" + ext)
+    DATA['simple_filenames'] = [pjoin(data_path, "simple" + ext)
                                 for ext in nib.streamlines.FORMATS.keys()]
-    DATA['complex_filenames'] = [pjoin(DATA_PATH, "complex" + ext)
+    DATA['complex_filenames'] = [pjoin(data_path, "complex" + ext)
                                  for ext in nib.streamlines.FORMATS.keys()]
 
     DATA['streamlines'] = [np.arange(1*3, dtype="f4").reshape((1, 3)),

nibabel/streamlines/tests/test_trk.py

@@ -6,6 +6,7 @@
 
 from nibabel.externals.six import BytesIO
 
+from nibabel.testing import data_path
 from nibabel.testing import clear_and_catch_warnings
 from nose.tools import assert_equal, assert_raises, assert_true
 from numpy.testing import assert_array_equal
@@ -18,24 +19,23 @@
 from ..trk import TrkFile
 from ..header import Field
 
-DATA_PATH = pjoin(os.path.dirname(__file__), 'data')
 DATA = {}
 
 
 def setup():
     global DATA
 
-    DATA['empty_trk_fname'] = pjoin(DATA_PATH, "empty.trk")
+    DATA['empty_trk_fname'] = pjoin(data_path, "empty.trk")
     # simple.trk contains only streamlines
-    DATA['simple_trk_fname'] = pjoin(DATA_PATH, "simple.trk")
+    DATA['simple_trk_fname'] = pjoin(data_path, "simple.trk")
     # standard.trk contains only streamlines
-    DATA['standard_trk_fname'] = pjoin(DATA_PATH, "standard.trk")
+    DATA['standard_trk_fname'] = pjoin(data_path, "standard.trk")
     # standard.LPS.trk contains only streamlines
-    DATA['standard_LPS_trk_fname'] = pjoin(DATA_PATH, "standard.LPS.trk")
+    DATA['standard_LPS_trk_fname'] = pjoin(data_path, "standard.LPS.trk")
 
     # complex.trk contains streamlines, scalars and properties
-    DATA['complex_trk_fname'] = pjoin(DATA_PATH, "complex.trk")
-    DATA['complex_trk_big_endian_fname'] = pjoin(DATA_PATH,
+    DATA['complex_trk_fname'] = pjoin(data_path, "complex.trk")
+    DATA['complex_trk_big_endian_fname'] = pjoin(data_path,
                                                  "complex_big_endian.trk")
 
     DATA['streamlines'] = [np.arange(1*3, dtype="f4").reshape((1, 3)),