From e65a84b9dd9dd4ac294fef8b09bec7f97bd2dec9 Mon Sep 17 00:00:00 2001
From: Paul Baltescu <pauldb89@gmail.com>
Date: Wed, 6 Mar 2013 20:01:08 +0000
Subject: Fixed unit test.

---
 extractor/target_phrase_extractor_test.cc | 5 +++--
 1 file 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.
 }
-- 
cgit v1.2.3