test save embeddings

vecto-ai · Apr 25, 2018 · 91658e5 · 91658e5
1 parent e3f5834
commit 91658e5
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/tests/test_embeddings.py b/tests/test_embeddings.py
@@ -26,13 +26,13 @@ def test_load(self):
         model.vocabulary.get_id("apple")
         print(model.name)
 
-#    def test_save(self):
-#        path = "./tests/data/embeddings/text/plain_with_file_header/"
-#        model = vsmlib.model.load_from_dir(path)
-#        path_save = "/tmp/vsmlib/saved"
-#        model.save_to_dir(path_save)
-#        model = vsmlib.model.load_from_dir(path_save)
-#        print(model.matrix.shape)
+    def test_save(self):
+        path = "./tests/data/embeddings/text/plain_with_file_header/"
+        model = vecto.embeddings.load_from_dir(path)
+        path_save = "/tmp/vsmlib/saved"
+        model.save_to_dir(path_save)
+        model = vecto.embeddings.load_from_dir(path_save)
+        print(model.matrix.shape)
 
 #    def test_load_numpy(self):
 #        path = "./tests/data/embeddings/npy/"