test: test using file objects

mammothb · Oct 18, 2024 · 14d2860 · 14d2860
1 parent f17a84c
commit 14d2860
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/test_symspellpy.py b/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)