check cell value type also

1 files changed, 3 insertions, 0 deletions

diff --git a/searchlib/src/test/java/com/yahoo/searchlib/tensor/TensorConformanceTest.java b/searchlib/src/test/java/com/yahoo/searchlib/tensor/TensorConformanceTest.java
index c28873684ee..61aec069c72 100644
--- a/searchlib/src/test/java/com/yahoo/searchlib/tensor/TensorConformanceTest.java
+++ b/searchlib/src/test/java/com/yahoo/searchlib/tensor/TensorConformanceTest.java
@@ -76,6 +76,9 @@ public class TensorConformanceTest {
             boolean equals = Tensor.equals(result, expect);
             if (!equals) {
                 System.out.println(count + " : Tensors not equal. Result: " + result.toString() + " Expected: " + expect.toString() + " -> expression \"" + expression + "\"");
+            } else if (! result.type().valueType().equals(expect.type().valueType())) {
+                System.out.println(count + " : Tensor cell value types not equal. Result: " + result.type() + " Expected: " + expect.type() + " -> expression \"" + expression + "\"");
+                equals = false;
             }
             return equals;