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
|
#include "search/weights.hh"
#define BOOST_TEST_MODULE WeightTest
#include <boost/test/unit_test.hpp>
#include <boost/test/floating_point_comparison.hpp>
namespace search {
namespace {
#define CHECK_WEIGHT(value, string) \
i = parsed.find(string); \
BOOST_REQUIRE(i != parsed.end()); \
BOOST_CHECK_CLOSE((value), i->second, 0.001);
BOOST_AUTO_TEST_CASE(parse) {
// These are not real feature weights.
Weights w("rarity=0 phrase-SGT=0 phrase-TGS=9.45117 lhsGrhs=0 lexical-SGT=2.33833 lexical-TGS=-28.3317 abstract?=0 LanguageModel=3 lexical?=1 glue?=5");
const boost::unordered_map<std::string, search::Score> &parsed = w.GetMap();
boost::unordered_map<std::string, search::Score>::const_iterator i;
CHECK_WEIGHT(0.0, "rarity");
CHECK_WEIGHT(0.0, "phrase-SGT");
CHECK_WEIGHT(9.45117, "phrase-TGS");
CHECK_WEIGHT(2.33833, "lexical-SGT");
BOOST_CHECK(parsed.end() == parsed.find("lm"));
BOOST_CHECK_CLOSE(3.0, w.LM(), 0.001);
CHECK_WEIGHT(-28.3317, "lexical-TGS");
CHECK_WEIGHT(5.0, "glue?");
}
BOOST_AUTO_TEST_CASE(dot) {
Weights w("rarity=0 phrase-SGT=0 phrase-TGS=9.45117 lhsGrhs=0 lexical-SGT=2.33833 lexical-TGS=-28.3317 abstract?=0 LanguageModel=3 lexical?=1 glue?=5");
BOOST_CHECK_CLOSE(9.45117 * 3.0, w.DotNoLM("phrase-TGS=3.0"), 0.001);
BOOST_CHECK_CLOSE(9.45117 * 3.0, w.DotNoLM("phrase-TGS=3.0 LanguageModel=10"), 0.001);
BOOST_CHECK_CLOSE(9.45117 * 3.0 + 28.3317 * 17.4, w.DotNoLM("rarity=5 phrase-TGS=3.0 LanguageModel=10 lexical-TGS=-17.4"), 0.001);
}
} // namespace
} // namespace search
|
