test for create vocab from file

undertherain · Apr 23, 2018 · 9c2b3cd · 9c2b3cd
1 parent 0d99636
commit 9c2b3cd
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/test_vocab.py b/test/test_vocab.py
@@ -1,7 +1,7 @@
 """Tests for vocabulary module."""
 
 import unittest
-from vsmlib.vocabulary import create_from_dir, Vocabulary
+from vsmlib.vocabulary import create_from_dir, create_from_file, Vocabulary
 
 path_text = "./test/data/corpora/plain"
 path_vocab = "./test/data/vocabs/plain"
@@ -15,6 +15,11 @@ def test_create_from_dir(self):
         assert vocab.get_id("the") >= 0
         vocab.save_to_dir("/tmp/vsmlib/vocab")
 
+    def test_create_from_file(self):
+        vocab = create_from_file("test/data/corpora/plain/sense_small.txt", min_frequency=10)
+        print("the:", vocab.get_id("the"))
+        assert vocab.get_id("the") >= 0
+
     def test_load_from_dir(self):
         vocab = Vocabulary()
         vocab.load(path_vocab)