diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2019-04-02 06:34:20 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2019-04-02 06:34:20 +0200 |
commit | 8ccf5a6afaa5a63530580f54d9146f23f61d6dff (patch) | |
tree | b7ca9c437caa776be7523ae59fed0a3acad3a3f3 /vespajlib | |
parent | 46b756d16093c0ef100b7f77f3f2dd661184aabb (diff) |
Test and ensure that serialization is symmetric.
Diffstat (limited to 'vespajlib')
-rw-r--r-- | vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java | 2 | ||||
-rw-r--r-- | vespajlib/src/test/java/com/yahoo/tensor/serialization/DenseBinaryFormatTestCase.java | 6 |
2 files changed, 7 insertions, 1 deletions
diff --git a/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java b/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java index 2537e7d8669..500c436516f 100644 --- a/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java +++ b/vespajlib/src/main/java/com/yahoo/tensor/serialization/DenseBinaryFormat.java @@ -117,7 +117,7 @@ public class DenseBinaryFormat implements BinaryFormat { serializedValueType = TensorType.ValueType.DOUBLE; break; case FLOAT_VALUE_TYPE: - serializedValueType = TensorType.ValueType.DOUBLE; + serializedValueType = TensorType.ValueType.FLOAT; break; default: throw new IllegalArgumentException("Received tensor value type '" + serializedValueType + "'. Only 0(double), or 1(float) are legal."); diff --git a/vespajlib/src/test/java/com/yahoo/tensor/serialization/DenseBinaryFormatTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/serialization/DenseBinaryFormatTestCase.java index b248603082f..e8b17812f32 100644 --- a/vespajlib/src/test/java/com/yahoo/tensor/serialization/DenseBinaryFormatTestCase.java +++ b/vespajlib/src/test/java/com/yahoo/tensor/serialization/DenseBinaryFormatTestCase.java @@ -69,6 +69,12 @@ public class DenseBinaryFormatTestCase { Arrays.toString(TypedBinaryFormat.encode(tensor))); } + @Test + public void testSerializationOfDifferentValueTypes() { + assertSerialization(TensorType.ValueType.DOUBLE, "tensor(x[],y[]):{{x:0,y:0}:2.0, {x:0,y:1}:3.0, {x:1,y:0}:4.0, {x:1,y:1}:5.0}"); + assertSerialization(TensorType.ValueType.FLOAT, "tensor(x[],y[]):{{x:0,y:0}:2.0, {x:0,y:1}:3.0, {x:1,y:0}:4.0, {x:1,y:1}:5.0}"); + } + private void assertSerialization(String tensorString) { assertSerialization(TensorType.ValueType.DOUBLE, Tensor.from(tensorString)); } |