File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed
Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change 44from __future__ import print_function
55
66import unittest
7- from decoder import *
7+ from models import decoder
88
99
1010class TestDecoders (unittest .TestCase ):
@@ -53,11 +53,13 @@ def setUp(self):
5353 self .beam_search_result = ['acdc' , "b'a" ]
5454
5555 def test_greedy_decoder_1 (self ):
56- bst_result = ctc_greedy_decoder (self .probs_seq1 , self .vocab_list )
56+ bst_result = decoder .ctc_greedy_decoder (self .probs_seq1 ,
57+ self .vocab_list )
5758 self .assertEqual (bst_result , self .greedy_result [0 ])
5859
5960 def test_greedy_decoder_2 (self ):
60- bst_result = ctc_greedy_decoder (self .probs_seq2 , self .vocab_list )
61+ bst_result = decoder .ctc_greedy_decoder (self .probs_seq2 ,
62+ self .vocab_list )
6163 self .assertEqual (bst_result , self .greedy_result [1 ])
6264
6365 def test_beam_search_decoder_1 (self ):
Original file line number Diff line number Diff line change 55from __future__ import print_function
66
77import unittest
8- import error_rate
8+ from utils import error_rate
99
1010
1111class TestParse (unittest .TestCase ):
You can’t perform that action at this time.
0 commit comments