“adb5d136bfdb35c9f090fabd4ba0d45af28c274d”上不存在“mobile/src/fpga/V1/bias_scale.h”
提交 d776ce9b 编写于 作者: X Xinghai Sun

Fix import errors in unitests.

上级 b2eb008a
...@@ -4,7 +4,7 @@ from __future__ import division ...@@ -4,7 +4,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import unittest import unittest
from decoder import * from models import decoder
class TestDecoders(unittest.TestCase): class TestDecoders(unittest.TestCase):
...@@ -53,15 +53,17 @@ class TestDecoders(unittest.TestCase): ...@@ -53,15 +53,17 @@ class TestDecoders(unittest.TestCase):
self.beam_search_result = ['acdc', "b'a"] self.beam_search_result = ['acdc', "b'a"]
def test_greedy_decoder_1(self): def test_greedy_decoder_1(self):
bst_result = ctc_greedy_decoder(self.probs_seq1, self.vocab_list) bst_result = decoder.ctc_greedy_decoder(self.probs_seq1,
self.vocab_list)
self.assertEqual(bst_result, self.greedy_result[0]) self.assertEqual(bst_result, self.greedy_result[0])
def test_greedy_decoder_2(self): def test_greedy_decoder_2(self):
bst_result = ctc_greedy_decoder(self.probs_seq2, self.vocab_list) bst_result = decoder.ctc_greedy_decoder(self.probs_seq2,
self.vocab_list)
self.assertEqual(bst_result, self.greedy_result[1]) self.assertEqual(bst_result, self.greedy_result[1])
def test_beam_search_decoder_1(self): def test_beam_search_decoder_1(self):
beam_result = ctc_beam_search_decoder( beam_result = decoder.ctc_beam_search_decoder(
probs_seq=self.probs_seq1, probs_seq=self.probs_seq1,
beam_size=self.beam_size, beam_size=self.beam_size,
vocabulary=self.vocab_list, vocabulary=self.vocab_list,
...@@ -69,7 +71,7 @@ class TestDecoders(unittest.TestCase): ...@@ -69,7 +71,7 @@ class TestDecoders(unittest.TestCase):
self.assertEqual(beam_result[0][1], self.beam_search_result[0]) self.assertEqual(beam_result[0][1], self.beam_search_result[0])
def test_beam_search_decoder_2(self): def test_beam_search_decoder_2(self):
beam_result = ctc_beam_search_decoder( beam_result = decoder.ctc_beam_search_decoder(
probs_seq=self.probs_seq2, probs_seq=self.probs_seq2,
beam_size=self.beam_size, beam_size=self.beam_size,
vocabulary=self.vocab_list, vocabulary=self.vocab_list,
...@@ -77,7 +79,7 @@ class TestDecoders(unittest.TestCase): ...@@ -77,7 +79,7 @@ class TestDecoders(unittest.TestCase):
self.assertEqual(beam_result[0][1], self.beam_search_result[1]) self.assertEqual(beam_result[0][1], self.beam_search_result[1])
def test_beam_search_decoder_batch(self): def test_beam_search_decoder_batch(self):
beam_results = ctc_beam_search_decoder_batch( beam_results = decoder.ctc_beam_search_decoder_batch(
probs_split=[self.probs_seq1, self.probs_seq2], probs_split=[self.probs_seq1, self.probs_seq2],
beam_size=self.beam_size, beam_size=self.beam_size,
vocabulary=self.vocab_list, vocabulary=self.vocab_list,
......
...@@ -5,7 +5,7 @@ from __future__ import division ...@@ -5,7 +5,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import unittest import unittest
import error_rate from utils import error_rate
class TestParse(unittest.TestCase): class TestParse(unittest.TestCase):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册