From 9bc5629f3bd98c17a59bc169d02c0cf90089cfdb Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Mon, 9 Jul 2018 09:31:22 +0100
Subject: [PATCH] TEST: Test fast wrapper
---
tests/test_wrappers.py | 15 +++++++++++++++
tests/test_wrapperutils.py | 5 -----
2 files changed, 15 insertions(+), 5 deletions(-)
diff --git a/tests/test_wrappers.py b/tests/test_wrappers.py
index 5dcca79c1..af2a95fb6 100644
--- a/tests/test_wrappers.py
+++ b/tests/test_wrappers.py
@@ -272,3 +272,18 @@ def test_fslmaths():
assert result.output[0] == expected
# TODO test LOAD output
+
+def test_fast():
+ with asrt.disabled(), run.dryrun(), mockFSLDIR() as fsldir:
+
+ cmd = op.join(fsldir, 'bin', 'fast')
+
+ result = fw.fast('input', 'myseg', n_classes=3)
+ expected = [cmd, '-v', '--out=myseg', '--class=3', 'input']
+
+ assert result.output[0] == ' '.join(expected)
+
+ result = fw.fast(('in1', 'in2', 'in3'), 'myseg', n_classes=3)
+ expected = [cmd, '-v', '--out=myseg', '--class=3', 'in1', 'in2', 'in3']
+
+ assert result.output[0] == ' '.join(expected)
diff --git a/tests/test_wrapperutils.py b/tests/test_wrapperutils.py
index 9a5d0d264..0399c8205 100644
--- a/tests/test_wrapperutils.py
+++ b/tests/test_wrapperutils.py
@@ -318,14 +318,9 @@ def test_fileOrThing_sequence():
if isinstance(arrs, six.string_types):
arrs = [arrs]
- print('Loading from files', arrs)
-
arrs = [np.loadtxt(a) for a in arrs]
-
res = np.sum(arrs, axis=0)
- print('result', res)
-
np.savetxt(out, res)
inputs = [np.random.randint(1, 10, (3, 3)) for i in range(4)]
--
GitLab