custom-dictionary

juji-io · Dec 15, 2021 · 66f973e · 66f973e
1 parent f9fe973
commit 66f973e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/project.clj b/project.clj
@@ -1,4 +1,4 @@
-(def version "0.2.0")
+(def version "0.2.1")
 
 (defproject org.clojars.huahaiy/symspell-clj version
   :description "SymSpell in Clojure"

diff --git a/src/symspell_clj/core.clj b/src/symspell_clj/core.clj
@@ -142,7 +142,9 @@
                               :or   {max-edit-distance 2
                                      prefix-length     10
                                      custom-dictionary []}}]
-   (let [unigram (read-unigram unigram-file)]
+   (let [unigram-file (or unigram-file "en_unigrams.txt")
+         bigram-file  (or bigram-file "en_bigrams.txt")
+         unigram      (read-unigram unigram-file)]
      (doseq [w custom-dictionary]
        (.put unigram w custom-word-default-freq))
      (->SpellChecker

diff --git a/test/symspell_clj/core_test.clj b/test/symspell_clj/core_test.clj
@@ -42,5 +42,8 @@
            [["what is fuji" 2]]))
     (sut/add-word sc "juji")
     (is (= (sut/lookup-compound sc "wht is juji")
-           [["what is juji" 1]]))
-    ))
+           [["what is juji" 1]]))))
+
+(deftest custom-dict-test
+  (let [sc (sut/new-spellchecker nil nil {:custom-dictionary ["juji"]})]
+    (is (= (sut/lookup sc "juji") [["juji" 0]]))))