test: add test cases for edit_distance_python()

tests/test_phonem_converter_score.py

@@ -2,10 +2,14 @@ import unittest
 
 import epitran
 
-from aip_trainer import pronunciationTrainer
+from aip_trainer import pronunciationTrainer, WordMetrics
 from aip_trainer.models import RuleBasedModels
 
 
+words_real = 'Ich habe sehr viel glück, am leben und gesund zu sein'
+words_transcribed = 'Ic hab zeh viel guck am und gesund tu sein'
+
+
 class TestPhonemConverter(unittest.TestCase):
 
     def test_english_ok(self):
@@ -26,8 +30,6 @@ trainer_SST_lambda = {'de': pronunciationTrainer.getTrainer("de")}
 class TestScore(unittest.TestCase):
 
     def test_exact_transcription(self):
-        words_real = 'Ich habe sehr viel glück, am leben und gesund zu sein'
-
         real_and_transcribed_words, _, _ = trainer_SST_lambda['de'].matchSampleAndRecordedWords(
             words_real, words_real)
 
@@ -37,8 +39,6 @@ class TestScore(unittest.TestCase):
         self.assertEqual(int(pronunciation_accuracy), 100)
 
     def test_incorrect_transcription(self):
-        words_real = 'Ich habe sehr viel glück, am leben und gesund zu sein'
-        words_transcribed = 'Ic hab zeh viel guck am und gesund tu sein'
 
         real_and_transcribed_words, _, _ = trainer_SST_lambda['de'].matchSampleAndRecordedWords(
             words_real, words_transcribed)
@@ -48,6 +48,64 @@ class TestScore(unittest.TestCase):
 
         self.assertEqual(int(pronunciation_accuracy), 71)
 
+    def test_edit_distance_python(self):
+        output = WordMetrics.edit_distance_python(words_real, words_transcribed)
+        self.assertEqual(output, int(14))
+
+    def test_edit_distance_python_same(self):
+        output = WordMetrics.edit_distance_python(words_real, words_real)
+        self.assertEqual(output, int(0))
+    
+    def test_edit_distance_python_empty(self):
+        output = WordMetrics.edit_distance_python("", "")
+        self.assertEqual(output, int(0))
+        output = WordMetrics.edit_distance_python(words_real, "")
+        self.assertEqual(output, int(53))
+        output = WordMetrics.edit_distance_python("", words_real)
+        self.assertEqual(output, int(53))
+
+    def test_edit_distance_python_not_printable_characters(self):
+        output = WordMetrics.edit_distance_python("Ich bin Alex\t a\0b! Hallo.", "Ich bi\tn Tom a\nc... Hallo")
+        self.assertEqual(output, int(12))
+
+    def test_edit_distance_python_survived(self):
+        # These tests are added based on the mutation test results to ensure the function is robust
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Add_Mul
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr")
+        self.assertEqual(output, int(5))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Add_Div
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel")
+        self.assertEqual(output, int(10))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Add_FloorDiv
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück")
+        self.assertEqual(output, int(16))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Add_Pow
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am")
+        self.assertEqual(output, int(20))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Sub_Mod
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am leben")
+        self.assertEqual(output, int(26))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Sub_RShift
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am leben und")
+        self.assertEqual(output, int(30))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Sub_LShift
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am leben und gesund")
+        self.assertEqual(output, int(37))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Sub_BitAnd
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am leben und gesund zu")
+        self.assertEqual(output, int(40))
+
+        # Test case where mutation survived for ReplaceBinaryOperator_Sub_BitXor
+        output = WordMetrics.edit_distance_python("Ich habe", "Ich habe sehr viel glück, am leben und gesund zu sein")
+        self.assertEqual(output, int(45))
 
 if __name__ == '__main__':
     unittest.main()