Fix tests

tests/test_stress.py

@@ -28,10 +28,10 @@ def test_stress_table():
         ("Привіт, як тебе звати?", "прив+іт, +як т+ебе зв+ати?"),
         ("АННА - український панк-рок гурт", "+анна - укра+їнський п+анк-р+ок г+урт"),
         ("Не тільки в Україні таке може бути.", "н+е т+ільки в укра+їні т+аке м+оже б+ути."),
-        ("Не тільки в +Укра+їні т+аке може бути.", "н+е т+ільки в ++укра++їні т+ак+е м+оже б+ути."),
+        ("Не тільки в +Укра+їні т+аке може бути.", "н+е т+ільки в +Укра+їні т+аке м+оже б+ути."),
         ("два + два", "дв+а + дв+а"),
         ("Н тльк в крн тк мж бт.", "н тльк в крн тк мж бт."),
     ]
 
     for item in examples:
-        assert stress_with_model(item[0]) == item[1]
+        assert sentence_to_stress(item[0], stress_function=stress_with_model) == item[1]