From 12e07e812ba87dcc41c4ccbf0fa26d6794c989d8 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauldmccarthy@gmail.com>
Date: Fri, 27 Sep 2024 17:29:55 +0100
Subject: [PATCH] TEST: Test saveLabelFile with classification probabilities
---
fsl/tests/test_fixlabels.py | 33 ++++++++++++++++++++++++++++-----
1 file changed, 28 insertions(+), 5 deletions(-)
diff --git a/fsl/tests/test_fixlabels.py b/fsl/tests/test_fixlabels.py
index 4155f2d3..2664e982 100644
--- a/fsl/tests/test_fixlabels.py
+++ b/fsl/tests/test_fixlabels.py
@@ -435,13 +435,8 @@ def test_loadLabelFile_probabilities():
assert lists_equal(gotprobs, expprobs)
-
-
-
-
def test_saveLabelFile():
-
labels = [['Label1', 'Label2', 'Label3'],
['Signal'],
['Noise'],
@@ -493,3 +488,31 @@ def test_saveLabelFile():
fixlabels.saveLabelFile(labels, fname, signalLabels=sigLabels)
with open(fname, 'rt') as f:
assert f.read().strip() == exp
+
+
+def test_saveLabelFile_probabilities():
+
+ labels = [['Label1', 'Label2', 'Label3'],
+ ['Signal'],
+ ['Noise'],
+ ['Label1'],
+ ['Unknown']]
+ probs = [0.1, 0.2, 0.3, 0.4, 0.5]
+
+ expected = tw.dedent("""
+ 1, Label1, Label2, Label3, True, 0.100000
+ 2, Signal, False, 0.200000
+ 3, Noise, True, 0.300000
+ 4, Label1, True, 0.400000
+ 5, Unknown, False, 0.500000
+ [1, 3, 4]
+ """).strip()
+
+ with tests.testdir() as testdir:
+ fname = op.join(testdir, 'fname.txt')
+
+ exp = '.\n{}'.format(expected)
+ fixlabels.saveLabelFile(labels, fname, probabilities=probs)
+ with open(fname, 'rt') as f:
+ got = f.read().strip()
+ assert got == exp
--
GitLab