Merge branch 'nrnrk-feature/load_dictionary_support_file_object' into…

… dev

symspellpy/symspellpy.py

@@ -314,19 +314,20 @@ def load_bigram_dictionary(
 
     def load_dictionary(
         self,
-        corpus: Union[Path, str],
+        corpus: Union[Path, str, IO[str]],
         term_index: int,
         count_index: int,
         separator: str = " ",
         encoding: Optional[str] = None,
-    ):
+    ) -> bool:
         """Loads multiple dictionary entries from a file of word/frequency count
         pairs.
 
         **NOTE**: Merges with any dictionary data already loaded.
 
         Args:
-            corpus: The path+filename of the file.
+            corpus: The path+filename of the file or a file object of the
+                dictionary.
             term_index: The column position of the word.
             count_index: The column position of the frequency count.
             separator: Separator characters between term(s) and count.
@@ -335,6 +336,11 @@ def load_dictionary(
         Returns:
             ``True`` if file loaded, or ``False`` if file not found.
         """
+        if not isinstance(corpus, (Path, str)):
+            return self._load_dictionary_stream(
+                corpus, term_index, count_index, separator
+            )
+
         corpus = Path(corpus)
         if not corpus.exists():
             logger.error(f"Dictionary file not found at {corpus}.")

tests/test_symspellpy.py

@@ -1,3 +1,4 @@
+from io import StringIO
 from pathlib import Path
 from unittest import TestCase
 
@@ -239,6 +240,18 @@ def test_load_dictionary_encoding(self, symspell_default):
         assert 1 == len(result)
         assert "АБИ" == result[0].term
 
+    def test_load_dictionary_from_string_io(self, symspell_default, dictionary_path):
+        with open(dictionary_path, "r") as f:
+            symspell_default.load_dictionary(StringIO(f.read()), 0, 1)
+            assert 82834 == symspell_default.word_count
+            assert 676094 == symspell_default.entry_count
+
+    def test_load_dictionary_from_text_io_wrapper(self, symspell_default, dictionary_path):
+        with open(dictionary_path, "r") as f:
+            symspell_default.load_dictionary(f, 0, 1)
+            assert 82834 == symspell_default.word_count
+            assert 676094 == symspell_default.entry_count
+
     def test_create_dictionary_invalid_path(self, symspell_default):
         with TestCase.assertLogs("symspellpy.symspellpy.logger", level="ERROR") as cm:
             assert not symspell_default.create_dictionary(INVALID_PATH)