Source Code for Module solvcon.tests.test_helper

 1  # -*- coding: UTF-8 -*- 
 2   
 3  from unittest import TestCase 
 4   
 5 -class TestHelper(TestCase): 
 6 -    def test_info(self): 
 7          import sys 
 8          from cStringIO import StringIO 
 9          from ..helper import info 
10          stdout = sys.stdout 
11          sys.stdout = StringIO() 
12          info('test message') 
13          self.assertEqual(sys.stdout.getvalue(), 'test message') 
14          sys.stdout = stdout 
15   
16 -    def test_platform(self): 
17          import sys 
18          from ..helper import iswin 
19          if sys.platform.startswith('win'): 
20              self.assertTrue(iswin()) 
21          else: 
22              self.assertFalse(iswin()) 
23   
24 -class TestPrinter(TestCase): 
25 -    def test_simple(self): 
26          from cStringIO import StringIO 
27          from ..helper import Printer 
28          stream = StringIO() 
29          p = Printer(stream) 
30          p('test message') 
31          self.assertEqual(stream.getvalue(), 'test message') 
32   
33 -    def test_prepost(self): 
34          from cStringIO import StringIO 
35          from ..helper import Printer 
36          stream = StringIO() 
37          p = Printer(stream, prefix='pre', postfix='post') 
38          p('test message') 
39          self.assertEqual(stream.getvalue(), 'pretest messagepost') 
40   
41 -    def test_multiple(self): 
42          from cStringIO import StringIO 
43          from ..helper import Printer 
44          stream1 = StringIO() 
45          stream2 = StringIO() 
46          p = Printer([stream1, stream2]) 
47          p('test message again') 
48          self.assertEqual(stream1.getvalue(), 'test message again') 
49          self.assertEqual(stream2.getvalue(), 'test message again') 
50   
51 -    def test_stdout(self): 
52          import sys 
53          from cStringIO import StringIO 
54          from ..helper import Printer 
55          stdout = sys.stdout 
56          sys.stdout = StringIO() 
57          p = Printer('sys.stdout') 
58          p('test message') 
59          self.assertEqual(sys.stdout.getvalue(), 'test message') 
60          sys.stdout = stdout 
61