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.
"""
def __init__(self, hide_prints=True):
"""Create object with hide_prints=False to disable print hiding"""
self.hide = hide_prints
def __enter__(self):
self._original_stdout = sys.stdout
devnull = open(os.devnull, "w")
sys.stdout = devnull
if self.hide:
self._original_stdout = sys.stdout
devnull = open(os.devnull, "w")
sys.stdout = devnull
def __exit__(self, exc_type, exc_val, exc_tb):
sys.stdout = self._original_stdout
if self.hide:
sys.stdout = self._original_stdout
class MockParser: