def test_rep(self): print('test rep') doc_1 = document.Document(r"sim_0.8/orig.txt") doc_2 = document.Document(r"sim_0.8/orig_0.8_rep.txt") ans = document.caculate_similarity(doc_1, doc_2) print(ans) self.assertGreaterEqual(ans, 0) self.assertLessEqual(ans, 1)
def run(): jieba.setLogLevel(logging.INFO) # get arguments from system terminal try: file_path_1 = sys.argv[1] file_path_2 = sys.argv[2] output_path = sys.argv[3] except: print("缺少参数!") return # build Document object try: doc_1 = document.Document(file_path_1) doc_2 = document.Document(file_path_2) except: return # caculate similarity try: cosine_similiarity = document.caculate_similarity(doc_1, doc_2) except: return # output try: output_file = open(output_path, "w") output_file.write("%.2f"%(cosine_similiarity)) output_file.close() except: print("%s打开失败 "%(sys.argv[3])) return print("%.2f"%(cosine_similiarity)) print("OK")