def save_load_clear(): from harvesttext import loadHT, saveHT para = "上港的武磊和恒大的郜林,谁是中国最好的前锋?那当然是武磊武球王了,他是射手榜第一,原来是弱点的单刀也有了进步" saveHT(ht, "ht_model1") ht2 = loadHT("ht_model1") print("cut with loaded model") print(ht2.seg(para)) ht2.clear() print("cut with cleared model") print(ht2.seg(para))
def test_save_load_clear(): sys.stdout, expected = open(get_current_function_name()+"_current","w"), open(get_current_function_name()+"_expected").read() from harvesttext import loadHT,saveHT para = "上港的武磊和恒大的郜林,谁是中国最好的前锋?那当然是武磊武球王了,他是射手榜第一,原来是弱点的单刀也有了进步" saveHT(ht,"ht_model1") ht2 = loadHT("ht_model1") print("cut with loaded model") print(ht2.seg(para)) ht2.clear() print("cut with cleared model") print(ht2.seg(para)) sys.stdout.close() assert open(get_current_function_name() + "_current").read() == expected