| |
|
|
| import os |
| from pocketsphinx import Decoder |
| import unittest |
|
|
| DATADIR = os.path.join(os.path.dirname(__file__), "../../test/data") |
|
|
|
|
| class TestDecoder(unittest.TestCase): |
| def _run_decode(self, decoder, expect_fail=False): |
| with open(os.path.join(DATADIR, "goforward.raw"), "rb") as fh: |
| buf = fh.read() |
| decoder.start_utt() |
| decoder.process_raw(buf, no_search=False, full_utt=True) |
| decoder.end_utt() |
| self._check_hyp(decoder.hyp().hypstr, decoder.seg(), expect_fail) |
|
|
| def _check_hyp(self, hyp, hypseg, expect_fail=False): |
| if expect_fail: |
| self.assertNotEqual(hyp, "go forward ten meters") |
| else: |
| self.assertEqual(hyp, "go forward ten meters") |
| words = [] |
| try: |
| for seg in hypseg: |
| if seg.word not in ("<s>", "</s>", "<sil>", "(NULL)"): |
| words.append(seg.word) |
| except AttributeError: |
| for word, start, end in hypseg: |
| if word not in ("<s>", "</s>", "<sil>", "(NULL)"): |
| words.append(word) |
| if expect_fail: |
| self.assertNotEqual(words, "go forward ten meters".split()) |
| else: |
| self.assertEqual(words, "go forward ten meters".split()) |
|
|
| def test_decoder(self): |
| decoder = Decoder() |
| print("log(1e-150) = ", decoder.logmath.log(1e-150)) |
| print("Pronunciation for word 'hello' is ", decoder.lookup_word("hello")) |
| self.assertEqual("HH AH L OW", decoder.lookup_word("hello")) |
| print("Pronunciation for word 'abcdf' is ", decoder.lookup_word("abcdf")) |
| self.assertEqual(None, decoder.lookup_word("abcdf")) |
|
|
| self._run_decode(decoder) |
|
|
| |
| print("Best 10 hypothesis: ") |
| for best, i in zip(decoder.nbest(), range(10)): |
| print(best.hypstr, best.score) |
|
|
| with open(os.path.join(DATADIR, "goforward.mfc"), "rb") as stream: |
| stream.read(4) |
| buf = stream.read(13780) |
| decoder.start_utt() |
| decoder.process_cep(buf, False, True) |
| decoder.end_utt() |
| hypothesis = decoder.hyp() |
| print( |
| "Best hypothesis: ", |
| hypothesis.hypstr, |
| " model score: ", |
| hypothesis.best_score, |
| " confidence: ", |
| hypothesis.prob, |
| ) |
| self.assertEqual("go forward ten meters", decoder.hyp().hypstr) |
|
|
| def test_reinit(self): |
| decoder = Decoder() |
| decoder.add_word("_forward", "F AO R W ER D", True) |
| self._run_decode(decoder) |
| |
| decoder.config["samprate"] = 48000 |
| decoder.reinit_feat() |
| self.assertEqual("F AO R W ER D", decoder.lookup_word("_forward")) |
| self._run_decode(decoder, expect_fail=True) |
| decoder.config["samprate"] = 16000 |
| |
| decoder.reinit() |
| self.assertEqual(None, decoder.lookup_word("_forward")) |
| self._run_decode(decoder) |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
