def test_correct(self): text = 'word' self.assertEqual('word', correct(text)) text = 'wor' self.assertEqual('for', correct(text)) text = 'percfeact' self.assertEqual('perfect', correct(text)) text = 'abcdefghijklmn' self.assertEqual('abcdefghijklmn', correct(text))
def test_equals(self): for pair in self.pairs: test_txt = pair[0] ref_txt = pair[1] output_txt = main.correct( test_txt ) # il testo corretto e' uncased per via di BERT uncased in uso self.assertEqual(output_txt, ref_txt, output_txt + ' != ' + ref_txt)
def test_not_equals(self): text = 'Pontedera is near Pisa' # il dizionario non riconosce la citta di Pontedera self.assertNotEqual(main.correct(text), text.lower()) text2 = 'The new Huauei has a 20 Mpx camera desined by Leica' # Huawei e Leica non predicibili da BERT self.assertNotEqual(main.correct(text2), text2.lower())
def assertNotNeedsCorrecting(self, msg): self.assertEqual(msg, correct(msg))
def assertCorrection(self, msg, correction): self.assertEqual(correct(msg), correction)