diff --git a/tests/test_wrappers.py b/tests/test_wrappers.py index 5dcca79c1e2b7c8bab6612493470d52f7f9aa80f..af2a95fb6df0c7254ebde60625ccc2264ebdaa2e 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 9a5d0d26407d29113bdb09308473d373c1bfdc9d..0399c8205797b02fb3c4dc06bc2ba706ce29c75d 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)]