File tree Expand file tree Collapse file tree 1 file changed +11
-12
lines changed Expand file tree Collapse file tree 1 file changed +11
-12
lines changed Original file line number Diff line number Diff line change @@ -31,33 +31,32 @@ def setUp(self):
31
31
"Tearing down\n " + \
32
32
"Test Finished"
33
33
self .runner = TestRunner ()
34
+ self .out = StringIO ()
35
+ self .saved_stdout = sys .stdout
36
+ sys .stdout = self .out
37
+
38
+ def tearDown (self ):
39
+ self .out .close ()
40
+ sys .stdout = self .saved_stdout
34
41
35
42
def test_tc1_output (self ):
36
- out = StringIO ()
37
- sys .stdout = out
38
43
self .tc1 .run ()
39
- output = out .getvalue ().strip ()
44
+ output = self . out .getvalue ().strip ()
40
45
self .assertEqual (output , self .average_result_tc1 )
41
46
42
47
def test_tc2_output (self ):
43
- out = StringIO ()
44
- sys .stdout = out
45
48
self .tc2 .run ()
46
- output = out .getvalue ().strip ()
49
+ output = self . out .getvalue ().strip ()
47
50
self .assertEqual (output , self .average_result_tc2 )
48
51
49
52
def test_tc3_output (self ):
50
- out = StringIO ()
51
- sys .stdout = out
52
53
self .tc3 .run ()
53
- output = out .getvalue ().strip ()
54
+ output = self . out .getvalue ().strip ()
54
55
self .assertEqual (output , self .average_result_tc3 )
55
56
56
57
def test_bunch_launch (self ):
57
- out = StringIO ()
58
- sys .stdout = out
59
58
self .runner .runAll ()
60
- output = out .getvalue ().strip ()
59
+ output = self . out .getvalue ().strip ()
61
60
self .assertEqual (output , str (self .average_result_tc1 + '\n \n ' +
62
61
self .average_result_tc2 + '\n \n ' +
63
62
self .average_result_tc3 ))
You can’t perform that action at this time.
0 commit comments