Skip to content

Commit ed16977

Browse files
author
Peter Amstutz
committed
Windows stdout tweak
1 parent 8a35fef commit ed16977

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/test_toolargparse.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -114,15 +114,20 @@ def execute(self,
114114
return {}, "success"
115115

116116
def test_dont_require_inputs():
117+
if sys.version_info[0] < 3:
118+
stream = BytesIO()
119+
else:
120+
stream = StringIO()
121+
117122
script = None
118123
try:
119124
script = NamedTemporaryFile(mode='w', delete=False)
120125
script.write(script_a)
121126
script.close()
122127

123-
assert main(argsl=["--debug", script.name, "--input", script.name], executor=NoopJobExecutor()) == 0
124-
assert main(argsl=["--debug", script.name], executor=NoopJobExecutor()) == 2
125-
assert main(argsl=["--debug", script.name], executor=NoopJobExecutor(), input_required=False) == 0
128+
assert main(argsl=["--debug", script.name, "--input", script.name], executor=NoopJobExecutor(), stdout=stream) == 0
129+
assert main(argsl=["--debug", script.name], executor=NoopJobExecutor(), stdout=stream) == 2
130+
assert main(argsl=["--debug", script.name], executor=NoopJobExecutor(), input_required=False, stdout=stream) == 0
126131

127132
except SystemExit as err:
128133
assert err.code == 0, name

0 commit comments

Comments
 (0)