結果 : how to write unit test for a function in python