Context manager to catch stdout.
| 13 | |
| 14 | |
| 15 | class CaptureOutput(List[str]): |
| 16 | """Context manager to catch stdout.""" |
| 17 | |
| 18 | def __enter__(self) -> "CaptureOutput": |
| 19 | self._original_stdout = sys.stdout |
| 20 | self._temp_stdout = StringIO() |
| 21 | sys.stdout = self._temp_stdout |
| 22 | return self |
| 23 | |
| 24 | def __exit__(self, *args: Any) -> None: |
| 25 | lines = self._temp_stdout.getvalue().splitlines() |
| 26 | self.extend(line.rstrip() for line in lines) |
| 27 | sys.stdout = self._original_stdout |
| 28 | |
| 29 | |
| 30 | def pprint_default(values: NodeValueList) -> List[str]: |
no outgoing calls
no test coverage detected