結果 : python mock return value based on input