diff options
| -rw-r--r-- | tests/test_utils.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tests/test_utils.py b/tests/test_utils.py index 43a05a9a..807bc0fd 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -119,11 +119,23 @@ def stdout_sim(): stdsim = cu.StdSim(sys.stdout) return stdsim +@pytest.fixture +def stringio_sim(): + import io + stdsim = cu.StdSim(io.StringIO(), echo=True) + return stdsim + + def test_stdsim_write_str(stdout_sim): my_str = 'Hello World' stdout_sim.write(my_str) assert stdout_sim.getvalue() == my_str +def test_stdsim_write_str_inner_no_buffer(stringio_sim): + my_str = 'Hello World' + stringio_sim.write(my_str) + assert stringio_sim.getvalue() == my_str + def test_stdsim_write_bytes(stdout_sim): b_str = b'Hello World' with pytest.raises(TypeError): |