def test_initiate_spellchecker_no_storage(self):
spelling_config = BotSpellingConfiguration()
spelling_config._load = True
spelling_config._classname = "programy.spelling.norvig.NorvigSpellingChecker"
client = TestClient()
storage_factory = client.storage_factory
spell_checker = SpellingChecker.initiate_spellchecker(spelling_config, storage_factory)
self.assertIsNotNone(spell_checker)
def test_initiate_spellchecker_no_classname(self):
spelling_config = BotSpellingConfiguration()
spelling_config._load = True
spelling_config._classname = None
storage_factory = None
spell_checker = SpellingChecker.initiate_spellchecker(
spelling_config, storage_factory)
self.assertIsNone(spell_checker)
def test_initiate_spellchecker_invalid_classname(self):
spelling_config = BotSpellingConfiguration()
spelling_config._load = True
spelling_config._classname = "programytest.spelling.test_base.InvalidSpellingChecker"
storage_factory = None
spell_checker = SpellingChecker.initiate_spellchecker(
spelling_config, storage_factory)
self.assertIsNone(spell_checker)
def test_check_spelling_before_true(self):
spelling_config = BotSpellingConfiguration()
spelling_config._load = True
spelling_config._classname = "programytest.spelling.test_base.MockSpellingChecker"
spelling_config._check_before = True
storage_factory = None
spell_checker = SpellingChecker.initiate_spellchecker(
spelling_config, storage_factory)
client = TestClient()
client_context = client.create_client_context("user1")
sentence = Sentence(client_context.brain.tokenizer, "Hello word")
spell_checker.check_spelling_before(client_context, sentence)
self.assertEqual(sentence.text(), "Hello World")
def test_check_spelling_and_retry_false(self):
spelling_config = BotSpellingConfiguration()
spelling_config._load = True
spelling_config._classname = "programytest.spelling.test_base.MockSpellingChecker"
spelling_config._check_and_retry = False
storage_factory = None
spell_checker = SpellingChecker.initiate_spellchecker(
spelling_config, storage_factory)
client = TestClient()
client_context = client.create_client_context("user1")
tokenizer = client_context.brain.tokenizer
client_context._brain = MockBrain()
sentence = Sentence(tokenizer, "Hello word")
response = spell_checker.check_spelling_and_retry(
client_context, sentence)
self.assertIsNone(response)