Python set_output_stream Examples

Example #1

    def run_dbt_and_capture(self, *args, **kwargs):
        try:
            initial_stdout = log_manager.stdout
            initial_stderr = log_manager.stderr
            stringbuf = io.StringIO()
            log_manager.set_output_stream(stringbuf)

            res = self.run_dbt(*args, **kwargs)
            stdout = stringbuf.getvalue()

        finally:
            log_manager.set_output_stream(initial_stdout, initial_stderr)

        return res, stdout

Example #2

 def tearDown(self):
     log_manager.set_output_stream(self.initial_stdout, self.initial_stderr)
     super().tearDown()

Example #3

 def setUp(self):
     super().setUp()
     self.initial_stdout = log_manager.stdout
     self.initial_stderr = log_manager.stderr
     self.stringbuf = io.StringIO()
     log_manager.set_output_stream(self.stringbuf)