Mere fix/dictionary-partly-yoficated-word-lookup into master

Author: unsektor

changelog.md

@@ -13,6 +13,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - [ ] feature: add lint option
 - [ ] feature: add replacement stats
 
+## [0.1.1] — 2025-03-02
+### Fixed
+
+- fix: dictionary word lookup for party yoficated words, e.g. `четырЕхзвЁздный`, `четырЁхзвЕздный`
+
 ## [0.1.0] — 2025-02-16
 
 `md.language.yofication` package initial version
@@ -71,4 +76,5 @@ Initial release
 
 </details>
 
+[0.1.1]: https://github.com/md-py/md.language.yofication/releases/tag/0.1.1
 [0.1.0]: https://github.com/md-py/md.language.yofication/releases/tag/0.1.0

lib/md/language/yofication/_yofication.py

@@ -120,10 +120,10 @@ def __init__(self, dictionary: DictionaryType) -> None:
             self._dictionary[ye_word.lower()] = yo_word.lower()
 
     def find(self, word: str) -> typing.Optional[str]:
-        return self._dictionary.get(word, None)
+        return self._dictionary.get(word.replace('ё', 'е'), None)
 
     def has(self, word: str) -> bool:
-        return word in self._dictionary
+        return word.replace('ё', 'е') in self._dictionary
 
 
 class RegularExpressionYoficate(YoficateInterface):

setup.py

@@ -16,7 +16,7 @@ def create_zip():
     create_zip()
     setuptools.setup(
         name='md.language.yofication',
-        version='0.1.0',
+        version='0.1.1',
         description='A Russian text yoficator (ёфикатор)',
         long_description=long_description,
         long_description_content_type='text/markdown',

tests/md/language/yofication/test_yofication.py

@@ -27,6 +27,32 @@ def test_has(self) -> None:
         assert mapping_dictionary.has(word='трехмерный') is True
         assert mapping_dictionary.has(word='четырехзвездный') is False
 
+    def test_find_partly_yoficated(self) -> None:
+        # arrange
+        dictionary = {'четырехзвездный': 'четырёхзвёздный'}
+
+        # act
+        mapping_dictionary = md.language.yofication.MappingDictionary(dictionary=dictionary)
+
+        # assert
+        assert mapping_dictionary.find(word='четырёхзвёздный') == 'четырёхзвёздный'
+        assert mapping_dictionary.find(word='четырехзвёздный') == 'четырёхзвёздный'
+        assert mapping_dictionary.find(word='четырёхзвездный') == 'четырёхзвёздный'
+        assert mapping_dictionary.find(word='четырехзвездный') == 'четырёхзвёздный'
+
+    def test_has_partly_yoficated(self) -> None:
+        # arrange
+        dictionary = {'четырехзвездный': 'четырёхзвёздный'}
+
+        # act
+        mapping_dictionary = md.language.yofication.MappingDictionary(dictionary=dictionary)
+
+        # assert
+        assert mapping_dictionary.has(word='четырёхзвёздный') is True
+        assert mapping_dictionary.has(word='четырехзвёздный') is True
+        assert mapping_dictionary.has(word='четырёхзвездный') is True
+        assert mapping_dictionary.has(word='четырехзвездный') is True
+
 
 class TestRegularExpressionYoficate(unittest.TestCase):
     def test_word(self) -> None: