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)