Add code coverage build

.github/workflows/hf-sync.yml

@@ -7,6 +7,33 @@ on:
   workflow_dispatch:
 
 jobs:
+  build:
+    name: Run pytest and checks
+    runs-on: ubuntu-latest
+    #strategy:
+    #  matrix:
+    #    python-version: ["3.7", "3.8", "3.9", "3.10"]
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python 3.9 #${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: 3.9 #${{ matrix.python-version }}
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          pip install pytest
+          pip install -r requirements-test.txt
+      #- name: Lint with flake8
+      #  run: |
+      #    # stop the build if there are Python syntax errors or undefined names
+      #    flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+      #    # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+      #    flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+      - name: Test with pytest
+        run: |
+          pytest
   sync-to-hub:
     runs-on: ubuntu-latest
     steps:

pytest.ini

@@ -1,2 +1,2 @@
 [pytest]
-addopts = --ignore=training
+addopts = --ignore=training --cov=ukrainian_tts

requirements-test.txt

@@ -1,3 +1,4 @@
 -r requirements.txt
 
-pytest==7.2.0
+pytest==7.2.0
+pytest-cov==4.0.0

tests/test_stress.py

@@ -3,7 +3,8 @@ from ukrainian_tts.stress import sentence_to_stress, stress_with_model
 
 def test_stress_table():
     examples = [
-        #("Бабин біб розцвів у дощ — Буде бабі біб у борщ.\n\nБоронила", "Б+абин б+іб розцв+ів +у д+ощ — Б+уде б+абі б+іб +у б+орщ.\n\nБоронила"),
+        ("Бабин біб розцвів у дощ — Буде бабі біб у борщ.\n\nБоронила", "Б+абин б+іб розцв+ів +у д+ощ — Б+уде б+абі б+іб +у б+орщ.  Борон+ила"),
+        ("Бабин біб розцвів у дощ — Буде бабі біб у борщ.,,Боронила", "Б+абин б+іб розцв+ів +у д+ощ — Б+уде б+абі б+іб +у б+орщ.,,Борон+ила"),
         ("Бобер на березі з бобренятами бублики пік.","Боб+ер н+а березі з бобрен+ятами б+ублики п+ік."),
         (
             "Кам'янець-Подільський - місто в Хмельницькій області України, центр Кам'янець-Подільської міської об'єднаної територіальної громади і Кам'янець-Подільського району.", 

ukrainian_tts/stress.py

@@ -46,6 +46,7 @@ def stress_dict(sentence: str):
 def sentence_to_stress(sentence: str, stress_function=stress_dict) -> str:
     # save custom stress positions
     all_stresses = []
+    sentence = sentence.replace("\n", " ")
     orig_words = sentence.split(" ")
     for i in range(0, len(orig_words)):
         if "+" in orig_words[i]:
@@ -74,4 +75,4 @@ def sentence_to_stress(sentence: str, stress_function=stress_dict) -> str:
         for stressed in all_stresses:
             words[stressed] = orig_words[stressed]
         return " ".join(words)
-    return new_stressed
+    return new_stressed