diff options
Diffstat (limited to 'extractor')
| -rw-r--r-- | extractor/target_phrase_extractor_test.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/extractor/target_phrase_extractor_test.cc b/extractor/target_phrase_extractor_test.cc index 51c753ee..a686d20b 100644 --- a/extractor/target_phrase_extractor_test.cc +++ b/extractor/target_phrase_extractor_test.cc @@ -80,8 +80,8 @@ TEST_F(TargetPhraseExtractorTest, TestExtractTightPhrasesTrue) { } TEST_F(TargetPhraseExtractorTest, TestExtractPhrasesTightPhrasesFalse) { - vector<string> target_words = {"a", "b", "c", "d", "e", "f"}; - vector<int> target_symbols = {20, 21, 22, 23, 24, 25, 26}; + vector<string> target_words = {"a", "b", "c", "d", "e", "f", "END_OF_LINE"}; + vector<int> target_symbols = {20, 21, 22, 23, 24, 25, 1}; EXPECT_CALL(*data_array, GetSentenceLength(0)).WillRepeatedly(Return(6)); EXPECT_CALL(*data_array, GetSentenceStart(0)).WillRepeatedly(Return(0)); @@ -110,6 +110,7 @@ TEST_F(TargetPhraseExtractorTest, TestExtractPhrasesTightPhrasesFalse) { vector<pair<Phrase, PhraseAlignment> > results = extractor->ExtractPhrases( target_gaps, target_low, 1, 5, source_indexes, 0); EXPECT_EQ(10, results.size()); + // TODO(pauldb): Finish unit test once it's clear how these alignments should // look like. } |