From 4f7aca981b6c049c5089e1071c2b9e410d6eaa35 Mon Sep 17 00:00:00 2001
From: Paul McCarthy <pauld.mccarthy@gmail.com>
Date: Tue, 18 Apr 2017 17:55:52 +0100
Subject: [PATCH] More fixes to fixlabels tests.
---
tests/test_fixlabels.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/tests/test_fixlabels.py b/tests/test_fixlabels.py
index f8d228ea5..04d1c7edc 100644
--- a/tests/test_fixlabels.py
+++ b/tests/test_fixlabels.py
@@ -333,9 +333,10 @@ def test_saveLabelFile():
fname = op.join(testdir, 'fname.txt')
# dirname=None, listBad=False
+ exp = '.\n{}'.format(expected)
fixlabels.saveLabelFile(labels, fname, listBad=False)
with open(fname, 'rt') as f:
- assert f.read().strip() == expected
+ assert f.read().strip() == exp
# dirname=something, listBad=False
dirname = 'Blob/a.ica'
@@ -346,13 +347,14 @@ def test_saveLabelFile():
# dirname=None, listBad=True
fixlabels.saveLabelFile(labels, fname)
- exp = '{}\n[1, 3, 4]'.format(expected)
+ exp = '.\n{}\n[1, 3, 4]'.format(expected)
with open(fname, 'rt') as f:
assert f.read().strip() == exp
# Custom signal labels
sigLabels = ['Label1']
exp = textwrap.dedent("""
+ .
1, Label1, Label2, Label3, False
2, Signal, True
3, Noise, True
--
GitLab