| 362 | setattr(sys, self.channel, self.buffer if self.suppress else self.tee) |
| 363 | |
| 364 | def __exit__(self, etype, value, traceback): |
| 365 | try: |
| 366 | if value is not None: |
| 367 | # If an error was raised, don't check anything else |
| 368 | return False |
| 369 | self.tee.flush() |
| 370 | setattr(sys, self.channel, self.orig_stream) |
| 371 | printed = self.buffer.getvalue() |
| 372 | for s in self.s: |
| 373 | if isinstance(s, _re_type): |
| 374 | assert s.search(printed), notprinted_msg.format(s.pattern, self.channel, printed) |
| 375 | else: |
| 376 | assert s in printed, notprinted_msg.format(s, self.channel, printed) |
| 377 | return False |
| 378 | finally: |
| 379 | self.tee.close() |
| 380 | |
| 381 | printed_msg = """Found {0!r} in printed output (on {1}): |
| 382 | ------- |