(self, channel)
| 25 | |
| 26 | class TeeTestCase(unittest.TestCase): |
| 27 | def tchan(self, channel): |
| 28 | trap = StringIO() |
| 29 | chan = StringIO() |
| 30 | text = "Hello" |
| 31 | |
| 32 | std_ori = getattr(sys, channel) |
| 33 | setattr(sys, channel, trap) |
| 34 | |
| 35 | tee = Tee(chan, channel=channel) |
| 36 | |
| 37 | print(text, end="", file=chan) |
| 38 | trap_val = trap.getvalue() |
| 39 | self.assertEqual(chan.getvalue(), text) |
| 40 | |
| 41 | tee.close() |
| 42 | |
| 43 | setattr(sys, channel, std_ori) |
| 44 | assert getattr(sys, channel) == std_ori |
| 45 | |
| 46 | def test(self): |
| 47 | for chan in ["stdout", "stderr"]: |