tst: move testing utilities

Author: mgxd

tests/test_regression.py

@@ -1,6 +1,4 @@
 """Testing conversion with conversion saved on datalad"""
-from tempfile import mkdtemp
-import os
 import os.path as op
 import json
 from glob import glob
@@ -13,37 +11,11 @@
 except ImportError:
 	have_datalad = False
 
-from heudiconv.cli.run import main as runner
 import heudiconv
+from heudiconv.cli.run import main as runner
+from .utils import fetch_data, gen_heudiconv_args
 
 
-def gen_heudiconv_args(datadir, outdir, subject, heuristic_file, xargs=None):
-	heuristic = op.realpath(op.join(
-								op.dirname(heudiconv.__file__),
-					        	'..',
-			        			'heuristics',
-			        			heuristic_file))
-
-	args = ["-d", op.join(datadir, 'sourcedata/{subject}/*/*/*.tgz'),
-			"-c", "dcm2niix",
-			"-o", outdir,
-			"-s", subject,
-			"-f", heuristic,
-			"--bids",
-			]
-	if xargs:
-		args += xargs
-
-	return args
-
-def fetch_data(tmpdir, subject):
-	"""Fetches some test dicoms"""
-	targetdir = os.path.join(tmpdir, 'QA')
-	api.install(path=targetdir,
-				source='///dbic/QA')
-	api.get('{}/sourcedata/{}'.format(targetdir, subject))
-	return targetdir
-
 @pytest.mark.parametrize('subject', ['sub-sid000143'])
 @pytest.mark.parametrize('heuristic', ['dbic_bids.py'])
 @pytest.mark.skipif(not have_datalad, reason="no datalad")

tests/utils.py

@@ -1,6 +1,35 @@
 import hashlib
+import os.path as op
 
 
 def md5sum(filename):
     with open(filename, 'rb') as f:
         return hashlib.md5(f.read()).hexdigest()
+
+def gen_heudiconv_args(datadir, outdir, subject, heuristic_file, xargs=None):
+	heuristic = op.realpath(op.join(
+								op.dirname(heudiconv.__file__),
+					        	'..',
+			        			'heuristics',
+			        			heuristic_file))
+
+	args = ["-d", op.join(datadir, 'sourcedata/{subject}/*/*/*.tgz'),
+			"-c", "dcm2niix",
+			"-o", outdir,
+			"-s", subject,
+			"-f", heuristic,
+			"--bids",
+			]
+	if xargs:
+		args += xargs
+
+	return args
+
+
+def fetch_data(tmpdir, subject):
+	"""Fetches some test dicoms using datalad"""
+	targetdir = os.path.join(tmpdir, 'QA')
+	api.install(path=targetdir,
+				source='///dbic/QA')
+	api.get('{}/sourcedata/{}'.format(targetdir, subject))
+	return targetdir