Skip to content
Snippets Groups Projects
Commit 4f7aca98 authored by Paul McCarthy's avatar Paul McCarthy
Browse files

More fixes to fixlabels tests.

parent 81166394
No related branches found
No related tags found
No related merge requests found
...@@ -333,9 +333,10 @@ def test_saveLabelFile(): ...@@ -333,9 +333,10 @@ def test_saveLabelFile():
fname = op.join(testdir, 'fname.txt') fname = op.join(testdir, 'fname.txt')
# dirname=None, listBad=False # dirname=None, listBad=False
exp = '.\n{}'.format(expected)
fixlabels.saveLabelFile(labels, fname, listBad=False) fixlabels.saveLabelFile(labels, fname, listBad=False)
with open(fname, 'rt') as f: with open(fname, 'rt') as f:
assert f.read().strip() == expected assert f.read().strip() == exp
# dirname=something, listBad=False # dirname=something, listBad=False
dirname = 'Blob/a.ica' dirname = 'Blob/a.ica'
...@@ -346,13 +347,14 @@ def test_saveLabelFile(): ...@@ -346,13 +347,14 @@ def test_saveLabelFile():
# dirname=None, listBad=True # dirname=None, listBad=True
fixlabels.saveLabelFile(labels, fname) 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: with open(fname, 'rt') as f:
assert f.read().strip() == exp assert f.read().strip() == exp
# Custom signal labels # Custom signal labels
sigLabels = ['Label1'] sigLabels = ['Label1']
exp = textwrap.dedent(""" exp = textwrap.dedent("""
.
1, Label1, Label2, Label3, False 1, Label1, Label2, Label3, False
2, Signal, True 2, Signal, True
3, Noise, True 3, Noise, True
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment