From 480211c39b3bf03d9b739eaada58bc213ff4c578 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauld.mccarthy@gmail.com>
Date: Mon, 14 Nov 2016 18:55:43 +0000
Subject: [PATCH] Test case for new removeDuplicates function
---
tests/test_fsl_utils_path.py | 83 +++++++++++++++++++++++++++++++++++-
1 file changed, 81 insertions(+), 2 deletions(-)
diff --git a/tests/test_fsl_utils_path.py b/tests/test_fsl_utils_path.py
index e733fd8fb..287cb4628 100644
--- a/tests/test_fsl_utils_path.py
+++ b/tests/test_fsl_utils_path.py
@@ -327,6 +327,85 @@ def test_shallowest():
assert fslpath.shallowest(path, suffixes) == output
+def test_removeDuplicates():
+
+ allowedExts = fslimage.ALLOWED_EXTENSIONS
+ groups = fslimage.FILE_GROUPS
+
+ # [(files_to_create,
+ # [([paths], [expected]),
+ # ...
+ # ]),
+ # ...
+ # ]
+ allTests = [
+ (['file.hdr', 'file.img'], [
+ (['file'], ['file']),
+ (['file.hdr'], ['file.hdr']),
+ (['file.img'], ['file.img']),
+ (['file.hdr', 'file.img'], ['file.hdr']),
+ (['file.img', 'file.hdr'], ['file.img']),
+ ]),
+
+ (['file.hdr', 'file.img', 'file.blob'], [
+ (['file'], ['file']),
+ (['file.hdr'], ['file.hdr']),
+ (['file.img'], ['file.img']),
+ (['file.hdr', 'file.img'], ['file.hdr']),
+ (['file.img', 'file.hdr'], ['file.img']),
+ ]),
+
+
+ (['file.hdr', 'file.img', 'file.nii'], [
+ (['file'], ['file']),
+ (['file.hdr'], ['file.hdr']),
+ (['file.img'], ['file.img']),
+ (['file.hdr', 'file.img'], ['file.hdr']),
+ (['file.img', 'file.hdr'], ['file.img']),
+ ]),
+
+
+ (['001.hdr', '001.img', '002.hdr', '002.img', '003.hdr', '003.img'], [
+ (['001', '002', '003'],
+ ['001', '002', '003']),
+
+ (['001.hdr', '002.hdr', '003.hdr'],
+ ['001.hdr', '002.hdr', '003.hdr']),
+ (['001.img', '002.img', '003.img'],
+ ['001.img', '002.img', '003.img']),
+
+ (['001.hdr', '001.img', '002.hdr', '002.img', '003.img'],
+ ['001.hdr', '002.hdr', '003.img']),
+ (['001.hdr', '001.img', '002.hdr', '002.img', '003.hdr', '003.img'],
+ ['001.hdr', '002.hdr', '003.hdr']),
+ (['001.img', '001.hdr', '002.img', '002.hdr', '003.img', '003.hdr'],
+ ['001.img', '002.img', '003.img']),
+ ])
+ ]
+
+ workdir = tempfile.mkdtemp()
+
+ try:
+ for files_to_create, tests in allTests:
+ for fn in files_to_create:
+ with open(op.join(workdir, fn), 'wt') as f:
+ f.write('{}\n'.format(fn))
+
+ for paths, expected in tests:
+
+ paths = [op.join(workdir, p) for p in paths]
+ result = fslpath.removeDuplicates(paths, allowedExts, groups)
+
+ assert result == [op.join(workdir, e) for e in expected]
+
+ for f in files_to_create:
+ os.remove(op.join(workdir, f))
+
+ finally:
+ shutil.rmtree(workdir)
+
+
+
def test_getFileGroup():
allowedExts = fslimage.ALLOWED_EXTENSIONS
@@ -571,8 +650,8 @@ def test_imcp_shouldPass(move=False):
src = op.join(indir, src)
- if move: fslpath.immv(src, imcp_dest, allowedExts=allowedExts, fileGroups=groups)
- else: fslpath.imcp(src, imcp_dest, allowedExts=allowedExts, fileGroups=groups)
+ if move: fslpath.immv(src, imcp_dest, allowedExts=allowedExts, fileGroups=groups, overwrite=True)
+ else: fslpath.imcp(src, imcp_dest, allowedExts=allowedExts, fileGroups=groups, overwrite=True)
copied = os.listdir(outdir)
copied = [f for f in copied if op.isfile(op.join(outdir, f))]
--
GitLab