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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/fastos/fastos.h>
#include <vespa/vespalib/testkit/test_kit.h>
#include <vespa/eval/eval/value_type.h>
#include <vespa/searchlib/fef/feature_type.h>
#include <vespa/searchlib/fef/featurenameparser.h>
#include <vespa/searchlib/features/rankingexpressionfeature.h>
#include <vespa/searchlib/fef/test/dummy_dependency_handler.h>
#include <vespa/searchlib/fef/test/indexenvironment.h>
using namespace search::features;
using namespace search::fef::test;
using namespace search::fef;
using namespace vespalib::eval;
using TypeMap = std::map<vespalib::string,vespalib::string>;
struct SetupResult {
IndexEnvironment index_env;
RankingExpressionBlueprint rank;
DummyDependencyHandler deps;
bool setup_ok;
SetupResult(const TypeMap &object_inputs, const vespalib::string &expression);
~SetupResult();
};
SetupResult::SetupResult(const TypeMap &object_inputs,
const vespalib::string &expression)
: index_env(), rank(), deps(rank), setup_ok(false)
{
rank.setName("self");
index_env.getProperties().add("self.rankingScript", expression);
for (const auto &input: object_inputs) {
deps.define_object_input(input.first, ValueType::from_spec(input.second));
}
setup_ok = rank.setup(index_env, {});
EXPECT_TRUE(!deps.accept_type_mismatch);
}
SetupResult::~SetupResult() {}
void verify_output_type(const TypeMap &object_inputs,
const vespalib::string &expression, const FeatureType &expect)
{
SetupResult result(object_inputs, expression);
EXPECT_TRUE(result.setup_ok);
EXPECT_EQUAL(1u, result.deps.output.size());
ASSERT_EQUAL(1u, result.deps.output_type.size());
if (expect.is_object()) {
EXPECT_EQUAL(expect.type(), result.deps.output_type[0].type());
} else {
EXPECT_TRUE(!result.deps.output_type[0].is_object());
}
}
void verify_setup_fail(const TypeMap &object_inputs,
const vespalib::string &expression)
{
SetupResult result(object_inputs, expression);
EXPECT_TRUE(!result.setup_ok);
EXPECT_EQUAL(0u, result.deps.output.size());
}
TEST("require that expression with only number inputs produce number output (compiled)") {
TEST_DO(verify_output_type({}, "a*b", FeatureType::number()));
}
TEST("require that expression with object input produces object output (interpreted)") {
TEST_DO(verify_output_type({{"b", "double"}}, "a*b", FeatureType::object(ValueType::double_type())));
}
TEST("require that ranking expression can resolve to concrete complex type") {
TEST_DO(verify_output_type({{"a", "tensor(x{},y{})"}, {"b", "tensor(y{},z{})"}}, "a*b",
FeatureType::object(ValueType::from_spec("tensor(x{},y{},z{})"))));
}
TEST("require that ranking expression can resolve to abstract complex type") {
TEST_DO(verify_output_type({{"a", "tensor"}}, "a*b", FeatureType::object(ValueType::from_spec("tensor"))));
}
TEST("require that ranking expression can resolve to 'any' type") {
TEST_DO(verify_output_type({{"a", "tensor(x{},y{})"}, {"b", "tensor"}}, "a*b",
FeatureType::object(ValueType::from_spec("any"))));
}
TEST("require that setup fails for incompatible types") {
TEST_DO(verify_setup_fail({{"a", "tensor(x{},y{})"}, {"b", "tensor(y[10],z{})"}}, "a*b"));
}
TEST_MAIN() { TEST_RUN_ALL(); }
|