Merge pull request #137 from sensein/test_mon

adding testmon as a dev dependency
sensein · Aug 9, 2024 · c96ee1f · c96ee1f
2 parents f1853bb + 715f9e4
commit c96ee1f
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -173,3 +173,8 @@ docs/
 # Speechbrain models
 pretrained_models
 src/tests/audio/tasks/pretrained_models/
+
+# testmon
+.testmondata
+.testmondata-shm
+.testmondata-wal
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -78,7 +78,7 @@ repos:
   hooks:
   - id: pytest
     name: pytest
-    entry: poetry run pytest
+    entry: poetry run pytest --testmon
     language: system
     types: [python]
     pass_filenames: false

diff --git a/pyproject.toml b/pyproject.toml
@@ -70,6 +70,7 @@ ruff = "~=0.3"
 codespell = "~=2.3"
 jupyter = "~=1.0"
 ipywidgets = "~=8.1"
+pytest-testmon = "~=2.1.1"
 
 [tool.poetry.group.docs]
 optional = true

diff --git a/src/tests/audio/tasks/text_to_speech_test.py b/src/tests/audio/tasks/text_to_speech_test.py
@@ -12,7 +12,7 @@
 
     @pytest.fixture
     def hf_model() -> HFModel:
-        """Fixture for HF model."""
+        """Fixture for the HF model."""
         return HFModel(path_or_uri="suno/bark-small", revision="main")
 
     def test_synthesize_texts(hf_model: HFModel) -> None: