1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#include <gtest/gtest.h>
#include <memory>
#include <vector>
#include "mocks/mock_alignment.h"
#include "mocks/mock_data_array.h"
#include "mocks/mock_rule_extractor_helper.h"
#include "mocks/mock_vocabulary.h"
#include "phrase.h"
#include "phrase_builder.h"
#include "target_phrase_extractor.h"
using namespace std;
using namespace ::testing;
namespace extractor {
namespace {
class TargetPhraseExtractorTest : public Test {
protected:
virtual void SetUp() {
data_array = make_shared<MockDataArray>();
alignment = make_shared<MockAlignment>();
vocabulary = make_shared<MockVocabulary>();
phrase_builder = make_shared<PhraseBuilder>(vocabulary);
helper = make_shared<MockRuleExtractorHelper>();
}
shared_ptr<MockDataArray> data_array;
shared_ptr<MockAlignment> alignment;
shared_ptr<MockVocabulary> vocabulary;
shared_ptr<PhraseBuilder> phrase_builder;
shared_ptr<MockRuleExtractorHelper> helper;
shared_ptr<TargetPhraseExtractor> extractor;
};
TEST_F(TargetPhraseExtractorTest, TestExtractTightPhrasesTrue) {
EXPECT_CALL(*data_array, GetSentenceLength(1)).WillRepeatedly(Return(5));
EXPECT_CALL(*data_array, GetSentenceStart(1)).WillRepeatedly(Return(3));
vector<string> target_words = {"a", "b", "c", "d", "e"};
vector<int> target_symbols = {20, 21, 22, 23, 24};
for (size_t i = 0; i < target_words.size(); ++i) {
EXPECT_CALL(*data_array, GetWordAtIndex(i + 3))
.WillRepeatedly(Return(target_words[i]));
EXPECT_CALL(*vocabulary, GetTerminalIndex(target_words[i]))
.WillRepeatedly(Return(target_symbols[i]));
EXPECT_CALL(*vocabulary, GetTerminalValue(target_symbols[i]))
.WillRepeatedly(Return(target_words[i]));
}
vector<pair<int, int> > links = {
make_pair(0, 0), make_pair(1, 3), make_pair(2, 2), make_pair(3, 1),
make_pair(4, 4)
};
EXPECT_CALL(*alignment, GetLinks(1)).WillRepeatedly(Return(links));
vector<int> gap_order = {1, 0};
EXPECT_CALL(*helper, GetGapOrder(_)).WillRepeatedly(Return(gap_order));
extractor = make_shared<TargetPhraseExtractor>(
data_array, alignment, phrase_builder, helper, vocabulary, 10, true);
vector<pair<int, int> > target_gaps = {make_pair(3, 4), make_pair(1, 2)};
vector<int> target_low = {0, 3, 2, 1, 4};
unordered_map<int, int> source_indexes = {{0, 0}, {2, 2}, {4, 4}};
vector<pair<Phrase, PhraseAlignment> > results = extractor->ExtractPhrases(
target_gaps, target_low, 0, 5, source_indexes, 1);
EXPECT_EQ(1, results.size());
vector<int> expected_symbols = {20, -2, 22, -1, 24};
EXPECT_EQ(expected_symbols, results[0].first.Get());
vector<string> expected_words = {"a", "c", "e"};
EXPECT_EQ(expected_words, results[0].first.GetWords());
vector<pair<int, int> > expected_alignment = {
make_pair(0, 0), make_pair(2, 2), make_pair(4, 4)
};
EXPECT_EQ(expected_alignment, results[0].second);
}
TEST_F(TargetPhraseExtractorTest, TestExtractPhrasesTightPhrasesFalse) {
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));
for (size_t i = 0; i < target_words.size(); ++i) {
EXPECT_CALL(*data_array, GetWordAtIndex(i))
.WillRepeatedly(Return(target_words[i]));
EXPECT_CALL(*vocabulary, GetTerminalIndex(target_words[i]))
.WillRepeatedly(Return(target_symbols[i]));
EXPECT_CALL(*vocabulary, GetTerminalValue(target_symbols[i]))
.WillRepeatedly(Return(target_words[i]));
}
vector<pair<int, int> > links = {make_pair(1, 1)};
EXPECT_CALL(*alignment, GetLinks(0)).WillRepeatedly(Return(links));
vector<int> gap_order = {0};
EXPECT_CALL(*helper, GetGapOrder(_)).WillRepeatedly(Return(gap_order));
extractor = make_shared<TargetPhraseExtractor>(
data_array, alignment, phrase_builder, helper, vocabulary, 10, false);
vector<pair<int, int> > target_gaps = {make_pair(2, 4)};
vector<int> target_low = {-1, 1, -1, -1, -1, -1};
unordered_map<int, int> source_indexes = {{1, 1}};
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.
}
} // namespace
} // namespace extractor
|
