Changeset cd552e in git for machine_learning/ml_python/tests/common/test_lookuptable.py

Timestamp:

Aug 12, 2019, 4:20:41 PM (5 years ago)

Author:

Murray Heymann <heymann.murray@…>

Branches:

(u'spielwiese', '17f1d200f27c5bd38f5dfc6e8a0879242279d1d8')

Children:

87d423b777b7762ac55fd9308018f3d7a12b9940

Parents:

b03e2f1e6cf3ff16ea956608cf7b560430904959

git-author:

Murray Heymann <heymann.murray@gmail.com>2019-08-12 16:20:41+02:00

git-committer:

Murray Heymann <heymann.murray@gmail.com>2019-08-12 16:20:52+02:00

Message:

Create dir in home for cache files for ml

File:

• machine_learning/ml_python/tests/common/test_lookuptable.py (modified) (4 diffs)

machine_learning/ml_python/tests/common/test_lookuptable.py

@@ -6 +6 @@
 from common.lookuptable import *
 from common.keyword_vector import count_occurances
-from common.constants import KEYWORDS_FILE, VECTORS_NPY, HELP_FILE_PATH
+from common.constants import KEYWORDS_FILE, VECTORS_NPY, HELP_FILE_PATH, \
+        HOME_DIR
 
 class TestLookuptableMethods(unittest.TestCase):
@@ -13 +14 @@
         os.system("rm -r " + HELP_FILE_PATH)
         fetch_tbz2_data()
-        fetch_tbz2_data()
         files = get_list_of_htm_files()
         self.assertGreater(len(files), 0)
 
     def test_extract_keywords(self):
-        extract_keywords()
+        #extract_keywords()
         self.assertTrue(os.path.isfile(KEYWORDS_FILE))
 
@@ -35 +35 @@
     def test_init_table_on_system(self):
         tbz2_path = os.path.join(HELP_FILE_PATH, "helpfiles.tbz2")
-        os.remove(tbz2_path)
-        os.remove(KEYWORDS_FILE)
-        os.remove(VECTORS_NPY)
+        os.system("rm -r " + HOME_DIR)
 
         init_table_on_system()
@@ -66 +64 @@
 
 if __name__ == '__main__':
+    if os.path.isdir(HOME_DIR):
+        os.system("rm -r " + HOME_DIR)
+    extract_keywords()
+    fetch_tbz2_data()
     #cProfile.run("unittest.main()")
     unittest.main()