79c2ce5dd4
* Refactor online recognizer. Make it easier to support other streaming models. Note that it is a breaking change for the Python API. `sherpa_onnx.OnlineRecognizer()` used before should be replaced by `sherpa_onnx.OnlineRecognizer.from_transducer()`.
29 lines
772 B
Python
29 lines
772 B
Python
# sherpa-onnx/python/tests/test_online_transducer_model_config.py
|
|
#
|
|
# Copyright (c) 2023 Xiaomi Corporation
|
|
#
|
|
# To run this single test, use
|
|
#
|
|
# ctest --verbose -R test_online_transducer_model_config_py
|
|
|
|
import unittest
|
|
|
|
import _sherpa_onnx
|
|
|
|
|
|
class TestOnlineTransducerModelConfig(unittest.TestCase):
|
|
def test_constructor(self):
|
|
config = _sherpa_onnx.OnlineTransducerModelConfig(
|
|
encoder="encoder.onnx",
|
|
decoder="decoder.onnx",
|
|
joiner="joiner.onnx",
|
|
)
|
|
assert config.encoder == "encoder.onnx", config.encoder
|
|
assert config.decoder == "decoder.onnx", config.decoder
|
|
assert config.joiner == "joiner.onnx", config.joiner
|
|
print(config)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|