diff options
author | Paul Baltescu <pauldb89@gmail.com> | 2013-03-06 20:01:08 +0000 |
---|---|---|
committer | Paul Baltescu <pauldb89@gmail.com> | 2013-03-06 20:01:08 +0000 |
commit | e65a84b9dd9dd4ac294fef8b09bec7f97bd2dec9 (patch) | |
tree | 8460685e7597c50b9b7ba1b5f1f6cd4fb1692522 | |
parent | 0c30e5f14d415ec9a399ca4af82cee1885e60f4f (diff) |
Fixed unit test.
-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. } |