diff --git a/tests/test_diceware.py b/tests/test_diceware.py index ce2bd0655b6a419a09874adbb20fb0020c4a6273..e7d364b70fed3216475e9d2b1ae1ca1ea3e55b6d 100644 --- a/tests/test_diceware.py +++ b/tests/test_diceware.py @@ -21,12 +21,14 @@ class FakeRandom(object): @pytest.fixture(scope="function") def argv_handler(request): - """This fixture restores sys.argv after tests. + """This fixture restores sys.argv and sys.stdin after tests. """ _argv_stored = sys.argv + _stdin_stored = sys.stdin def teardown(): sys.argv = _argv_stored + sys.stdin = _stdin_stored request.addfinalizer(teardown)