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