Source Code for Module solvcon.io.tests.test_vtk

 1  # -*- coding: UTF-8 -*- 
 2   
 3  import os 
 4  from unittest import TestCase 
 5  from ...testing import loadfile 
 6  from .. import gambit 
 7   
 8 -class WriteTest(TestCase): 
 9      __test__ = False 
10      blk = None 
11      str_neu_vtk = None 
12   
13 -    def assertLines(self, str1, str2): 
14          """ 
15          Compare two input strings line by line. 
16          """ 
17          import sys 
18          lines1 = str1.splitlines() 
19          lines2 = str2.splitlines() 
20          self.assertEqual(len(lines1), len(lines2)) 
21          iswin = sys.platform.startswith('win') 
22          nlines = len(lines1) 
23          i = 0 
24          while i < nlines: 
25              oline = lines1[i] 
26              nline = lines2[i] 
27              # compare only non-float lines. 
28              if not iswin or 'e' not in oline.lower(): 
29                  self.assertEqual(oline, nline) 
30              i += 1 
31    
32 -    def test_legacy(self): 
33          import StringIO 
34          from .. import vtk 
35          outf = StringIO.StringIO() 
36          writer = vtk.VtkLegacyUstGridWriter(self.blk) 
37          writer.write(outf) 
38          str_blk_vtk = outf.getvalue() 
39          # compare new result with old result line by line. 
40          self.assertLines(self.str_neu_vtk, str_blk_vtk) 
41   
42 -class TestWriteSingle(WriteTest): 
43      __test__ = True 
44      blk = gambit.GambitNeutral(loadfile('sample.neu')).toblock( 
45          fpdtype='float32') 
46      str_neu_vtk = loadfile('sample.neu.single.vtk') 
47   
48 -class TestWriteDouble(WriteTest): 
49      __test__ = True 
50      blk = gambit.GambitNeutral(loadfile('sample.neu')).toblock( 
51          fpdtype='float64') 
52      str_neu_vtk = loadfile('sample.neu.double.vtk') 
53