From f47861f1f38e644ad17d6acfd2872af8bcb7d090 Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Tue, 8 Oct 2019 09:36:57 +0200 Subject: Add test --- .../yahoo/tensor/serialization/JsonFormatTestCase.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'vespajlib') diff --git a/vespajlib/src/test/java/com/yahoo/tensor/serialization/JsonFormatTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/serialization/JsonFormatTestCase.java index 4c44cbbf5c7..32d62903af5 100644 --- a/vespajlib/src/test/java/com/yahoo/tensor/serialization/JsonFormatTestCase.java +++ b/vespajlib/src/test/java/com/yahoo/tensor/serialization/JsonFormatTestCase.java @@ -51,6 +51,24 @@ public class JsonFormatTestCase { assertEquals(tensor, decoded); } + @Test + public void testMixedTensor() { + Tensor.Builder builder = Tensor.Builder.of(TensorType.fromSpec("tensor(x{},y[2])")); + builder.cell().label("x", "a").label("y", "0").value(1.0); + builder.cell().label("x", "a").label("y", "1").value(2.0); + builder.cell().label("x", "b").label("y", "0").value(3.0); + builder.cell().label("x", "b").label("y", "1").value(4.0); + Tensor tensor = builder.build(); + byte[] json = JsonFormat.encode(tensor); + assertEquals("{\"cells\":[" + + "{\"address\":{\"x\":\"a\"},\"values\":[1.0,2.0]}," + + "{\"address\":{\"x\":\"b\"},\"values\":[3.0,4.0]}" + + "]}", + new String(json, StandardCharsets.UTF_8)); + Tensor decoded = JsonFormat.decode(tensor.type(), json); + assertEquals(tensor, decoded); + } + @Test public void testDenseTensorInDenseForm() { Tensor.Builder builder = Tensor.Builder.of(TensorType.fromSpec("tensor(x[2],y[3])")); -- cgit v1.2.3