Added model tests

Author: istupakov

.github/workflows/python-package.yml

@@ -8,7 +8,7 @@ permissions:
 
 on:
   push:
-    branches: [ "main" ]
+    branches: [ "main", "tests" ]
   pull_request:
     branches: [ "main" ]
 

tests/onnx_asr/test_recognize.py

@@ -5,13 +5,24 @@
 import onnx_asr.utils
 from onnx_asr.adapters import TextResultsAsrAdapter
 
+models = [
+    "gigaam-v2-ctc",
+    "gigaam-v2-rnnt",
+    "nemo-fastconformer-ru-ctc",
+    "nemo-fastconformer-ru-rnnt",
+    "alphacep/vosk-model-ru",
+    "alphacep/vosk-model-small-ru",
+    "whisper-base",
+    "onnx-community/whisper-tiny",
+]
+
 
 @pytest.fixture(scope="module")
 def model(request: pytest.FixtureRequest) -> TextResultsAsrAdapter:
-    return onnx_asr.load_model(request.param)
+    return onnx_asr.load_model(request.param, quantization="int8" if request.param != "onnx-community/whisper-tiny" else "uint8")
 
 
-@pytest.mark.parametrize("model", ["alphacep/vosk-model-small-ru", "onnx-community/whisper-tiny", "whisper-base"], indirect=True)
+@pytest.mark.parametrize("model", models, indirect=True)
 def test_supported_only_mono_audio_error(model: TextResultsAsrAdapter) -> None:
     rng = np.random.default_rng(0)
     waveform = rng.random((1 * 16_000, 2), dtype=np.float32)
@@ -20,7 +31,7 @@ def test_supported_only_mono_audio_error(model: TextResultsAsrAdapter) -> None:
         model.recognize(waveform)
 
 
-@pytest.mark.parametrize("model", ["alphacep/vosk-model-small-ru", "onnx-community/whisper-tiny", "whisper-base"], indirect=True)
+@pytest.mark.parametrize("model", models, indirect=True)
 def test_wrong_sample_rate_error(model: TextResultsAsrAdapter) -> None:
     rng = np.random.default_rng(0)
     waveform = rng.random((1 * 16_000), dtype=np.float32)
@@ -29,7 +40,7 @@ def test_wrong_sample_rate_error(model: TextResultsAsrAdapter) -> None:
         model.recognize(waveform, sample_rate=24_000)  # type: ignore
 
 
-@pytest.mark.parametrize("model", ["alphacep/vosk-model-small-ru", "onnx-community/whisper-tiny", "whisper-base"], indirect=True)
+@pytest.mark.parametrize("model", models, indirect=True)
 def test_recognize(model: TextResultsAsrAdapter) -> None:
     rng = np.random.default_rng(0)
     waveform = rng.random((1 * 16_000), dtype=np.float32)