summaryrefslogtreecommitdiff
path: root/unittest_utils.py
diff options
context:
space:
mode:
authorScott <[email protected]>2022-02-02 13:47:36 -0800
committerScott <[email protected]>2022-02-02 13:47:36 -0800
commit7ff2af6fe7bffea90dc4a31c93140c189917c659 (patch)
tree34cb138c3a572037b5bc1d1b52ad9d25a2604ccb /unittest_utils.py
parent6ba90a1f30f1c0cf4df12fcd0c62181f29bc3668 (diff)
Let's be explicit with asserts; there was a bug in histogram
caused by assert foo when foo was an int with valid valid 0.
Diffstat (limited to 'unittest_utils.py')
-rw-r--r--unittest_utils.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/unittest_utils.py b/unittest_utils.py
index ba9ca28..b259c6b 100644
--- a/unittest_utils.py
+++ b/unittest_utils.py
@@ -286,12 +286,12 @@ class RecordStdout(object):
def __enter__(self) -> Callable[[], tempfile.SpooledTemporaryFile]:
self.recorder = contextlib.redirect_stdout(self.destination)
- assert self.recorder
+ assert self.recorder is not None
self.recorder.__enter__()
return lambda: self.destination
def __exit__(self, *args) -> Optional[bool]:
- assert self.recorder
+ assert self.recorder is not None
self.recorder.__exit__(*args)
self.destination.seek(0)
return None
@@ -315,12 +315,12 @@ class RecordStderr(object):
def __enter__(self) -> Callable[[], tempfile.SpooledTemporaryFile]:
self.recorder = contextlib.redirect_stderr(self.destination) # type: ignore
- assert self.recorder
+ assert self.recorder is not None
self.recorder.__enter__()
return lambda: self.destination
def __exit__(self, *args) -> Optional[bool]:
- assert self.recorder
+ assert self.recorder is not None
self.recorder.__exit__(*args)
self.destination.seek(0)
return None