summaryrefslogtreecommitdiff
path: root/extractor/target_phrase_extractor_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'extractor/target_phrase_extractor_test.cc')
-rw-r--r--extractor/target_phrase_extractor_test.cc5
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.
}