From aa858ad8ce87b527a2ed940620a80f64e7ff3ae4 Mon Sep 17 00:00:00 2001 From: Paul McCarthy <pauldmccarthy@gmail.com> Date: Tue, 30 Jul 2019 15:34:32 +0100 Subject: [PATCH] TEST: Extra vest tests --- tests/test_vest.py | 32 ++++++++++++++++++++++++++++++-- 1 file changed, 30 insertions(+), 2 deletions(-) diff --git a/tests/test_vest.py b/tests/test_vest.py index df74b5b8c..407c502ef 100644 --- a/tests/test_vest.py +++ b/tests/test_vest.py @@ -111,11 +111,37 @@ testfile4Colours = np.array([ testfile5 = """Obviously not a VEST file""" +testfile6 = """%BeginInstance +/SavedInstanceClassName /ClassLUT +/PseudoColourmap [ +<-color{0.0,1.0,5.0}-> +<-color{1.0,2.0,4.0}-> +<-color{2.0,3.0,3.0}-> +] +>> +%%EndInstance +%%EOF +""" + +testfile7 = """%%BeginInstance +/SavedInstanceClassName /ClassLUT +/PseudoColourmap [ +<-color{0.0,1.0,5.0}-> +<-color{1.0,2.0,4.0}-> +<-color{2.0,3.0,3.0}-> +] +>> +%%EndInstance +%%EOF +""" + def _createFiles(testdir): - names = ['testfile1', 'testfile2', 'testfile3', 'testfile4', 'testfile5'] - texts = [ testfile1, testfile2, testfile3, testfile4, testfile5] + names = ['testfile1', 'testfile2', 'testfile3', 'testfile4', 'testfile5', + 'testfile6', 'testfile7'] + texts = [ testfile1, testfile2, testfile3, testfile4, testfile5, + testfile6, testfile7] for name, text in zip(names, texts): filename = op.join(testdir, '{}.txt'.format(name)) @@ -137,6 +163,8 @@ def test_looksLikeVestLutFile(): assert vest.looksLikeVestLutFile(op.join(testdir, 'testfile3.txt')) assert vest.looksLikeVestLutFile(op.join(testdir, 'testfile4.txt')) assert not vest.looksLikeVestLutFile(op.join(testdir, 'testfile5.txt')) + assert vest.looksLikeVestLutFile(op.join(testdir, 'testfile6.txt')) + assert vest.looksLikeVestLutFile(op.join(testdir, 'testfile7.txt')) finally: shutil.rmtree(testdir) -- GitLab