def check_flake8(self) -> None: """Run flake8 on files and/or repo""" errors: List[str] = [] with utils.buffered(stdout=errors, mangle=self._strip_lines): self.flake8_app.run_checks() self.flake8_app.report() if errors: self.error("flake8", errors)
def test_util_buffered_stdout(): stdout = [] with utils.buffered(stdout=stdout): print("test1") print("test2") sys.stdout.write("test3\n") sys.stderr.write("error0\n") assert stdout == ["test1", "test2", "test3"]
def test_util_buffered_stderr(): stderr = [] with utils.buffered(stderr=stderr): print("test1") print("test2") sys.stdout.write("test3\n") sys.stderr.write("error0\n") sys.stderr.write("error1\n") assert stderr == ["error0", "error1"]
def test_util_buffered_no_stdout_stderr(): with pytest.raises(utils.BufferUtilError): with utils.buffered(): pass