def test_mute_restore_custom_sigint(change_sigint_handler): assert signal.getsignal(signal.SIGINT) is custom_sigint_handler before_exit._mute_sigint() assert ( signal.getsignal(signal.SIGINT) is before_exit._print_still_working_message ) before_exit._restore_sigint() assert signal.getsignal(signal.SIGINT) is custom_sigint_handler
def test_restore_sigint(capsys, change_sigint_handler): before_exit._restore_sigint() with pytest.raises(KeyboardInterrupt): os.kill(pid, signal.SIGINT)