diff --git a/tests/test_diceware.py b/tests/test_diceware.py index b95b1c4ba44106a4680c9c89db3c72afd32a2ff0..eecc2f1dda379a3c4edafe6a0affbb2f298243a4 100644 --- a/tests/test_diceware.py +++ b/tests/test_diceware.py @@ -175,14 +175,13 @@ class TestDicewareModule(object): assert not ('\n' in out[:-1]) # we get one line assert len(out) > 5 # we get at least some chars - def test_main_help(self, capsys): + def test_main_help(self, argv_handler, capsys): # we can get help + sys.argv = ['diceware', '--help'] with pytest.raises(SystemExit) as exc_info: - main(['--help']) + main() assert exc_info.value.code == 0 out, err = capsys.readouterr() - out = out.replace( - os.path.basename(sys.argv[0]), 'diceware') assert out == ( 'usage: diceware [-h] [-n NUM] [-c | --no-caps] [-s NUM] [INFILE]\n' '\n'