Python Catch Keyboard Input Test