Python keyboard test of library pygame