diff options
| author | Scott <[email protected]> | 2022-02-02 13:47:36 -0800 |
|---|---|---|
| committer | Scott <[email protected]> | 2022-02-02 13:47:36 -0800 |
| commit | 7ff2af6fe7bffea90dc4a31c93140c189917c659 (patch) | |
| tree | 34cb138c3a572037b5bc1d1b52ad9d25a2604ccb /unittest_utils.py | |
| parent | 6ba90a1f30f1c0cf4df12fcd0c62181f29bc3668 (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.py | 8 |
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 |
