把測試切換到py.test，並根據結果修正一些BUG