From 837cd6f890a518ac5bd3355900a0d7f479fe5328 Mon Sep 17 00:00:00 2001 From: ChenXin Date: Tue, 14 May 2019 23:15:55 +0800 Subject: [PATCH] =?UTF-8?q?cache=5Fresults=20=E5=8F=82=E6=95=B0=E7=9A=84?= =?UTF-8?q?=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/core/test_utils.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/test/core/test_utils.py b/test/core/test_utils.py index b846a32d..6d70d97c 100644 --- a/test/core/test_utils.py +++ b/test/core/test_utils.py @@ -140,7 +140,7 @@ class TestCache(unittest.TestCase): try: start_time = time.time() embed, vocab, d = process_data_1('test/data_for_tests/word2vec_test.txt', 'test/data_for_tests/cws_train', - cache_filepath='test/demo_overwrite.pkl') + _cache_fp='test/demo_overwrite.pkl') end_time = time.time() pre_time = end_time - start_time with open('test/demo_overwrite.pkl', 'rb') as f: @@ -150,7 +150,7 @@ class TestCache(unittest.TestCase): self.assertListEqual(embed[i].tolist(), _embed[i].tolist()) start_time = time.time() embed, vocab, d = process_data_1('test/data_for_tests/word2vec_test.txt', 'test/data_for_tests/cws_train', - cache_filepath='test/demo_overwrite.pkl') + _cache_fp='test/demo_overwrite.pkl') end_time = time.time() read_time = end_time - start_time print("Read using {:.3f}, while prepare using:{:.3f}".format(read_time, pre_time)) @@ -162,7 +162,7 @@ class TestCache(unittest.TestCase): try: start_time = time.time() embed, vocab, d = process_data_1('test/data_for_tests/word2vec_test.txt', 'test/data_for_tests/cws_train', - refresh=True) + _refresh=True) end_time = time.time() pre_time = end_time - start_time with open('test/demo1.pkl', 'rb') as f: @@ -172,7 +172,7 @@ class TestCache(unittest.TestCase): self.assertListEqual(embed[i].tolist(), _embed[i].tolist()) start_time = time.time() embed, vocab, d = process_data_1('test/data_for_tests/word2vec_test.txt', 'test/data_for_tests/cws_train', - refresh=True) + _refresh=True) end_time = time.time() read_time = end_time - start_time print("Read using {:.3f}, while prepare using:{:.3f}".format(read_time, pre_time))