1 from contextlib
import redirect_stderr, redirect_stdout
2 from io
import StringIO
4 from pybind11_tests
import iostream
as m
8 msg =
"I've been redirected to Python, I hope!"
10 stdout, stderr = capsys.readouterr()
14 m.captured_output_default(msg)
15 stdout, stderr = capsys.readouterr()
20 stdout, stderr = capsys.readouterr()
27 msg =
"I've been redirected to Python, I hope!"
28 msg = msg * (1024 //
len(msg) + 1)
30 m.captured_output_default(msg)
31 stdout, stderr = capsys.readouterr()
38 msg =
"" + msg * (1024 //
len(msg) + 1)
40 m.captured_output_default(msg)
41 stdout, stderr = capsys.readouterr()
48 msg =
"1" + msg * (1024 //
len(msg) + 1)
50 m.captured_output_default(msg)
51 stdout, stderr = capsys.readouterr()
58 msg =
"" + msg * (1024 //
len(msg) + 1)
60 m.captured_output_default(msg)
61 stdout, stderr = capsys.readouterr()
68 msg =
"1" + msg * (1024 //
len(msg) + 1)
70 m.captured_output_default(msg)
71 stdout, stderr = capsys.readouterr()
78 msg =
"12" + msg * (1024 //
len(msg) + 1)
80 m.captured_output_default(msg)
81 stdout, stderr = capsys.readouterr()
88 msg =
"" + msg * (1024 //
len(msg) + 1)
90 m.captured_output_default(msg)
91 stdout, stderr = capsys.readouterr()
98 msg =
"1" + msg * (1024 //
len(msg) + 1)
100 m.captured_output_default(msg)
101 stdout, stderr = capsys.readouterr()
108 msg =
"12" + msg * (1024 //
len(msg) + 1)
110 m.captured_output_default(msg)
111 stdout, stderr = capsys.readouterr()
118 msg =
"123" + msg * (1024 //
len(msg) + 1)
120 m.captured_output_default(msg)
121 stdout, stderr = capsys.readouterr()
127 msg =
"I've been redirected to Python, I hope!"
129 stdout, stderr = capsys.readouterr()
136 m.captured_output(
"a")
137 m.captured_output(
"b")
138 assert capture ==
"ab"
142 msg =
"(not flushed)"
145 with m.ostream_redirect():
146 m.noisy_function(msg, flush=
False)
147 stdout, stderr = capfd.readouterr()
150 m.noisy_function(msg2, flush=
True)
151 stdout, stderr = capfd.readouterr()
152 assert stdout == msg + msg2
154 m.noisy_function(msg, flush=
False)
156 stdout, stderr = capfd.readouterr()
161 msg =
"Something that should not show up in log"
163 with redirect_stdout(stream):
165 stdout, stderr = capfd.readouterr()
168 assert not stream.getvalue()
171 with redirect_stdout(stream):
172 m.captured_output(msg)
173 stdout, stderr = capfd.readouterr()
176 assert stream.getvalue() == msg
180 msg =
"Something that should not show up in log"
182 with redirect_stderr(stream):
184 stdout, stderr = capfd.readouterr()
187 assert not stream.getvalue()
190 with redirect_stderr(stream):
192 stdout, stderr = capfd.readouterr()
195 assert stream.getvalue() == msg
200 with redirect_stdout(stream):
201 m.captured_output(
"a")
203 m.captured_output(
"c")
205 stdout, stderr = capfd.readouterr()
206 assert stdout ==
"bd"
207 assert stream.getvalue() ==
"ac"
211 m.captured_dual(
"a",
"b")
212 stdout, stderr = capsys.readouterr()
218 msg =
"Should not be in log!"
220 with redirect_stdout(stream):
222 stdout, stderr = capfd.readouterr()
224 assert not stream.getvalue()
227 with redirect_stdout(stream), m.ostream_redirect():
229 stdout, stderr = capfd.readouterr()
231 assert stream.getvalue() == msg
234 with redirect_stdout(stream):
236 stdout, stderr = capfd.readouterr()
238 assert not stream.getvalue()
246 with redirect_stderr(stream), m.ostream_redirect(stdout=
False):
249 stdout, stderr = capfd.readouterr()
252 assert stream.getvalue() == msg2
261 with redirect_stdout(stream), redirect_stderr(stream2), m.ostream_redirect():
264 stdout, stderr = capfd.readouterr()
267 assert stream.getvalue() == msg
268 assert stream2.getvalue() == msg2
272 with m.ostream_redirect(stdout=
True, stderr=
False):
278 threads.append(m.TestThread())