From 416f596b150ec159717bfd2f9b2ef70e4d4cd3dd Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Sat, 14 Jan 2023 18:41:49 +0100 Subject: Support direct tensor rendering --- .../src/test/java/com/yahoo/document/json/JsonWriterTestCase.java | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'document/src/test/java/com') diff --git a/document/src/test/java/com/yahoo/document/json/JsonWriterTestCase.java b/document/src/test/java/com/yahoo/document/json/JsonWriterTestCase.java index edf410b312e..eab33afc3e4 100644 --- a/document/src/test/java/com/yahoo/document/json/JsonWriterTestCase.java +++ b/document/src/test/java/com/yahoo/document/json/JsonWriterTestCase.java @@ -452,16 +452,16 @@ public class JsonWriterTestCase { doc.setFieldValue(tensorField, new TensorFieldValue(tensor)); assertEqualJson(asDocument(docId, "{ \"tensorfield\": {\"type\":\"tensor(x[3])\", \"cells\":[{\"address\":{\"x\":\"0\"},\"value\":1.0},{\"address\":{\"x\":\"1\"},\"value\":2.0},{\"address\":{\"x\":\"2\"},\"value\":3.0}]} }"), - writeDocument(doc, false)); + writeDocument(doc, false, false)); assertEqualJson(asDocument(docId, "{ \"tensorfield\": {\"type\":\"tensor(x[3])\", \"values\":[1.0, 2.0, 3.0] } }"), - writeDocument(doc, true)); + writeDocument(doc, true, false)); } - private byte[] writeDocument(Document doc, boolean tensorShortForm) throws IOException { + private byte[] writeDocument(Document doc, boolean tensorShortForm, boolean tensorDirectValues) throws IOException { ByteArrayOutputStream out = new ByteArrayOutputStream(); JsonFactory factory = new JsonFactory(); JsonGenerator generator = factory.createGenerator(out); - JsonWriter writer = new JsonWriter(generator, tensorShortForm); + JsonWriter writer = new JsonWriter(generator, tensorShortForm, tensorDirectValues); writer.write(doc); return out.toByteArray(); } -- cgit v1.2.3