fix: dictionary word lookup for party yoficated words, e.g. четырЕхзвЁздный, четырЁхзвЕздный

Author: unsektor

changelog.md

@@ -13,6 +13,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - [ ] feature: add lint option
 - [ ] feature: add replacement stats
 
+### Fixed
+
+- fix: dictionary word lookup for party yoficated words, e.g. `четырЕхзвЁздный`, `четырЁхзвЕздный`
+
 ## [0.1.0] — 2025-02-16
 
 `md.language.yofication` package initial version

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):

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: