blob: 1ba0e9b98e6da88efbc28b27eef2e7a27d55936b (
plain) (
blame)
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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.document.datatypes;
import com.yahoo.tensor.Tensor;
import com.yahoo.tensor.TensorType;
import org.junit.Test;
import java.util.Optional;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
/**
* @author geirst
*/
public class TensorFieldValueTestCase {
private static TensorFieldValue createFieldValue(String tensorString) {
return new TensorFieldValue(Tensor.from(tensorString));
}
@Test
public void requireThatDifferentFieldValueTypesAreNotEqual() {
assertFalse(createFieldValue("{{x:0}:2.0}").equals(new IntegerFieldValue(5)));
}
@Test
public void requireThatDifferentTensorTypesWithEmptyValuesAreNotEqual() {
TensorFieldValue field1 = new TensorFieldValue(new TensorType.Builder().mapped("x").build());
TensorFieldValue field2 = new TensorFieldValue(new TensorType.Builder().indexed("y").build());
assertFalse(field1.equals(field2));
}
@Test
public void requireThatDifferentTensorValuesAreNotEqual() {
TensorFieldValue field1 = createFieldValue("{{x:0}:2.0}");
TensorFieldValue field2 = createFieldValue("{{x:0}:3.0}");
assertFalse(field1.equals(field2));
assertFalse(field1.equals(new TensorFieldValue(TensorType.empty)));
}
@Test
public void requireThatSameTensorValueIsEqual() {
Tensor tensor = Tensor.from("{{x:0}:2.0}");
TensorFieldValue field1 = new TensorFieldValue(tensor);
TensorFieldValue field2 = new TensorFieldValue(tensor);
assertTrue(field1.equals(field1));
assertTrue(field1.equals(field2));
assertTrue(field1.equals(createFieldValue("{{x:0}:2.0}")));
}
@Test
public void requireThatToStringWorks() {
TensorFieldValue field1 = createFieldValue("{{x:0}:2.0}");
assertEquals("tensor(x{}):{0:2.0}", field1.toString());
TensorFieldValue field2 = new TensorFieldValue(TensorType.fromSpec("tensor(x{})"));
assertEquals("null", field2.toString());
}
@Test
public void requireThatSerializationHappens() {
TensorFieldValue orig = createFieldValue("{{x:0}:2.0}");
assertEquals("tensor(x{}):{0:2.0}", orig.toString());
assertTrue(orig.getWrappedValue() != null);
Optional<byte[]> serializedBytes = orig.getSerializedTensor();
assertTrue(serializedBytes.isPresent());
Optional<byte[]> otherSerialized = orig.getSerializedTensor();
assertTrue(otherSerialized.isPresent());
assertTrue(serializedBytes.get() == otherSerialized.get());
TensorFieldValue copy = new TensorFieldValue();
assertTrue(copy.getSerializedTensor().isEmpty());
copy.assignSerializedTensor(serializedBytes.get());
assertFalse(copy.getSerializedTensor().isEmpty());
otherSerialized = copy.getSerializedTensor();
assertTrue(serializedBytes.get() == otherSerialized.get());
copy = new TensorFieldValue();
assertTrue(copy.getTensorType().isEmpty());
copy.assignSerializedTensor(serializedBytes.get());
assertFalse(copy.getTensorType().isEmpty());
assertEquals(copy.getTensorType().get(), orig.getTensorType().get());
copy = new TensorFieldValue();
assertTrue(copy.getTensor().isEmpty());
copy.assignSerializedTensor(serializedBytes.get());
assertFalse(copy.getTensor().isEmpty());
copy = new TensorFieldValue();
assertTrue(copy.getTensor().isEmpty());
copy.assignSerializedTensor(serializedBytes.get());
assertFalse(copy.getTensor().isEmpty());
assertEquals(copy.getTensor().get(), orig.getTensor().get());
copy = new TensorFieldValue();
assertTrue(copy.getWrappedValue() == null);
copy.assignSerializedTensor(serializedBytes.get());
assertFalse(copy.getWrappedValue() == null);
assertEquals(copy.getWrappedValue(), orig.getWrappedValue());
}
}
|
