| import contextlib |
| import io |
| import sys |
| import unittest |
| |
| |
| class CfgCheckerTestBase(unittest.TestCase): |
| dummy_base_var = 0 |
| |
| def _safe_import_module(self, _str): |
| _import_msg = "" |
| _module = None |
| |
| try: |
| _module = __import__(_str) |
| except ImportError as e: |
| _import_msg = e.message |
| |
| return _import_msg, _module |
| |
| @contextlib.contextmanager |
| def redirect_output(self): |
| save_stdout = sys.stdout |
| save_stderr = sys.stderr |
| sys.stdout = io.BytesIO() |
| sys.stderr = io.BytesIO() |
| yield |
| sys.stdout = save_stdout |
| sys.stderr = save_stderr |