marcel/pylot
3
0
Fork 0
Code Issues 7 Pull Requests 1 Releases Wiki Activity

[add] HidePrint can be configured to skip hiding for quick debugging of tests

Browse Source
This commit is contained in:
Darius Arnold 2018-07-20 15:04:48 +02:00
parent 7fe5d0f7a3
commit 1b815a27b8
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/test_autopickstation/test_autopickstation.py
View File

@ -14,13 +14,20 @@ class HidePrints:
""" """
Used to hide all standard output the Function to be tested have, since it clutters the test results. Used to hide all standard output the Function to be tested have, since it clutters the test results.
""" """
def __init__(self, hide_prints=True):
"""Create object with hide_prints=False to disable print hiding"""
self.hide = hide_prints
def __enter__(self): def __enter__(self):
self._original_stdout = sys.stdout if self.hide:
devnull = open(os.devnull, "w") self._original_stdout = sys.stdout
sys.stdout = devnull devnull = open(os.devnull, "w")
sys.stdout = devnull
def __exit__(self, exc_type, exc_val, exc_tb): def __exit__(self, exc_type, exc_val, exc_tb):
sys.stdout = self._original_stdout if self.hide:
sys.stdout = self._original_stdout
class MockParser: class MockParser: