結果 : write test cases in python