Merge pull request #23708 from mamccorm/pyenchant-tests

Add additional tests

Author: rawlingsj

enchant2.yaml

@@ -59,10 +59,39 @@ test:
   environment:
     contents:
       packages:
-        - python3
+        - enchant2-dev
+        - hunspell-dev
+        - hunspell-dictionaries-all
         - py3-pip
+        - python3
   pipeline:
     - runs: |
-        pip install pyenchant
-        # this fails without the enchant libraries installed
-        python -c "import enchant"
+        pip3 install spello pyenchant
+    - runs: |
+        python -c '
+        import enchant
+        from spello.model import SpellCorrectionModel
+
+        d = enchant.Dict("en_US")
+
+        word = "recieve"
+        assert not d.check(word), f"{word} should be misspelled"
+        suggestions = d.suggest(word)
+        expected_suggestions = ["receive", "relieve", "reverie"]
+        assert set(expected_suggestions).issubset(set(suggestions)), f"Suggestions for {word} are incorrect. Got: {suggestions}"
+
+        sp = SpellCorrectionModel(language="en")
+        sp.train(["I want to play cricket", "this is a text corpus"])
+
+        sentence = "i wnt to plai kricket"
+        correction = sp.spell_correct(sentence)
+        expected_correction = {
+            "original_text": "i wnt to plai kricket",
+            "spell_corrected_text": "i want to play cricket",
+            "correction_dict": {
+                "wnt": "want",
+                "plai": "play",
+                "kricket": "cricket"
+            }
+        }
+        assert correction == expected_correction, f"Correction for {sentence} is incorrect. Expected: {expected_correction}, but got: {correction}"'