| """ | |
| test.py is a test script for the module. | |
| """ | |
| # %% | |
| from neuspell import BertChecker | |
| checker = BertChecker() | |
| checker.from_pretrained(".") | |
| # %% | |
| test1 = "I luk foward to receving your reply" | |
| test1_result = checker.correct(test1) # correct a string | |
| print(f"Original: {test1}", f"Corrected: {test1_result}", sep="\t") | |
| test2 = [test1, "were did wendigo goe boating?", "I am a good at math"] | |
| test2_result = checker.correct_strings(test2) # correct a list of strings | |
| print(f"Original: {test1}", f"Corrected: {test2_result}", sep="\t") | |
| # %% | |