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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
// Unit tests for tensor_fieldvalue.
#include <vespa/log/log.h>
LOG_SETUP("fieldvalue_test");
#include <vespa/document/base/exceptions.h>
#include <vespa/document/datatype/tensor_data_type.h>
#include <vespa/document/fieldvalue/tensorfieldvalue.h>
#include <vespa/eval/eval/simple_value.h>
#include <vespa/eval/eval/tensor_spec.h>
#include <vespa/eval/eval/value.h>
#include <vespa/vespalib/testkit/testapp.h>
using namespace document;
using vespalib::eval::SimpleValue;
using vespalib::eval::TensorSpec;
using vespalib::eval::ValueType;
namespace
{
TensorDataType xSparseTensorDataType(ValueType::from_spec("tensor(x{})"));
TensorDataType xySparseTensorDataType(ValueType::from_spec("tensor(x{},y{})"));
vespalib::eval::Value::UP createTensor(const TensorSpec &spec) {
return SimpleValue::from_spec(spec);
}
std::unique_ptr<vespalib::eval::Value>
makeSimpleTensor()
{
return SimpleValue::from_spec(TensorSpec("tensor(x{},y{})").
add({{"x", "4"}, {"y", "5"}}, 7));
}
FieldValue::UP clone(FieldValue &fv) {
auto ret = FieldValue::UP(fv.clone());
EXPECT_NOT_EQUAL(ret.get(), &fv);
EXPECT_EQUAL(*ret, fv);
EXPECT_EQUAL(fv, *ret);
return ret;
}
}
TEST("require that TensorFieldValue can be assigned tensors and cloned") {
TensorFieldValue noTensorValue(xySparseTensorDataType);
TensorFieldValue emptyTensorValue(xySparseTensorDataType);
TensorFieldValue twoCellsTwoDimsValue(xySparseTensorDataType);
emptyTensorValue = createTensor(TensorSpec("tensor(x{},y{})"));
twoCellsTwoDimsValue = createTensor(TensorSpec("tensor(x{},y{})")
.add({{"x", ""}, {"y", "3"}}, 3)
.add({{"x", "4"}, {"y", "5"}}, 7));
EXPECT_NOT_EQUAL(noTensorValue, emptyTensorValue);
EXPECT_NOT_EQUAL(noTensorValue, twoCellsTwoDimsValue);
EXPECT_NOT_EQUAL(emptyTensorValue, noTensorValue);
EXPECT_NOT_EQUAL(emptyTensorValue, twoCellsTwoDimsValue);
EXPECT_NOT_EQUAL(twoCellsTwoDimsValue, noTensorValue);
EXPECT_NOT_EQUAL(twoCellsTwoDimsValue, emptyTensorValue);
FieldValue::UP noneClone = clone(noTensorValue);
FieldValue::UP emptyClone = clone(emptyTensorValue);
FieldValue::UP twoClone = clone(twoCellsTwoDimsValue);
EXPECT_NOT_EQUAL(*noneClone, *emptyClone);
EXPECT_NOT_EQUAL(*noneClone, *twoClone);
EXPECT_NOT_EQUAL(*emptyClone, *noneClone);
EXPECT_NOT_EQUAL(*emptyClone, *twoClone);
EXPECT_NOT_EQUAL(*twoClone, *noneClone);
EXPECT_NOT_EQUAL(*twoClone, *emptyClone);
TensorFieldValue twoCellsTwoDimsValue2(xySparseTensorDataType);
twoCellsTwoDimsValue2 = createTensor(TensorSpec("tensor(x{},y{})")
.add({{"x", ""}, {"y", "3"}}, 3)
.add({{"x", "4"}, {"y", "5"}}, 7));
EXPECT_NOT_EQUAL(*noneClone, twoCellsTwoDimsValue2);
EXPECT_NOT_EQUAL(*emptyClone, twoCellsTwoDimsValue2);
EXPECT_EQUAL(*twoClone, twoCellsTwoDimsValue2);
}
TEST("require that TensorFieldValue::toString works")
{
TensorFieldValue tensorFieldValue(xSparseTensorDataType);
EXPECT_EQUAL("{TensorFieldValue: null}", tensorFieldValue.toString());
tensorFieldValue = createTensor(TensorSpec("tensor(x{})").add({{"x", "a"}}, 3));
EXPECT_EQUAL("{TensorFieldValue: spec(tensor(x{})) {\n [a]: 3\n}}", tensorFieldValue.toString());
}
TEST("require that wrong tensor type for special case assign throws exception")
{
TensorFieldValue tensorFieldValue(xSparseTensorDataType);
EXPECT_EXCEPTION(tensorFieldValue = makeSimpleTensor(),
document::WrongTensorTypeException,
"WrongTensorTypeException: Field tensor type is 'tensor(x{})' but other tensor type is 'tensor(x{},y{})'");
}
TEST("require that wrong tensor type for copy assign throws exception")
{
TensorFieldValue tensorFieldValue(xSparseTensorDataType);
TensorFieldValue simpleTensorFieldValue(xySparseTensorDataType);
simpleTensorFieldValue = makeSimpleTensor();
EXPECT_EXCEPTION(tensorFieldValue = simpleTensorFieldValue,
document::WrongTensorTypeException,
"WrongTensorTypeException: Field tensor type is 'tensor(x{})' but other tensor type is 'tensor(x{},y{})'");
}
TEST("require that wrong tensor type for assignDeserialized throws exception")
{
TensorFieldValue tensorFieldValue(xSparseTensorDataType);
EXPECT_EXCEPTION(tensorFieldValue.assignDeserialized(makeSimpleTensor()),
document::WrongTensorTypeException,
"WrongTensorTypeException: Field tensor type is 'tensor(x{})' but other tensor type is 'tensor(x{},y{})'");
}
TEST_MAIN() { TEST_RUN_ALL(); }
|