結果 : how to input test cases in python