Skip to content

Commit bff71d9

Browse files
committed
setUp and tearDown implementation for stdout redirect
1 parent 2457f14 commit bff71d9

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

tests/test_facade.py

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -31,33 +31,32 @@ def setUp(self):
3131
"Tearing down\n" + \
3232
"Test Finished"
3333
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
3441

3542
def test_tc1_output(self):
36-
out = StringIO()
37-
sys.stdout = out
3843
self.tc1.run()
39-
output = out.getvalue().strip()
44+
output = self.out.getvalue().strip()
4045
self.assertEqual(output, self.average_result_tc1)
4146

4247
def test_tc2_output(self):
43-
out = StringIO()
44-
sys.stdout = out
4548
self.tc2.run()
46-
output = out.getvalue().strip()
49+
output = self.out.getvalue().strip()
4750
self.assertEqual(output, self.average_result_tc2)
4851

4952
def test_tc3_output(self):
50-
out = StringIO()
51-
sys.stdout = out
5253
self.tc3.run()
53-
output = out.getvalue().strip()
54+
output = self.out.getvalue().strip()
5455
self.assertEqual(output, self.average_result_tc3)
5556

5657
def test_bunch_launch(self):
57-
out = StringIO()
58-
sys.stdout = out
5958
self.runner.runAll()
60-
output = out.getvalue().strip()
59+
output = self.out.getvalue().strip()
6160
self.assertEqual(output, str(self.average_result_tc1 + '\n\n' +
6261
self.average_result_tc2 + '\n\n' +
6362
self.average_result_tc3))

0 commit comments

Comments
 (0)