Test subtype methods

1 files changed, 24 insertions, 0 deletions

diff --git a/vespajlib/src/test/java/com/yahoo/tensor/TensorTypeTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/TensorTypeTestCase.java
index 738697d4521..ba541ab2cd6 100644
--- a/vespajlib/src/test/java/com/yahoo/tensor/TensorTypeTestCase.java
+++ b/vespajlib/src/test/java/com/yahoo/tensor/TensorTypeTestCase.java
@@ -102,6 +102,30 @@ public class TensorTypeTestCase {
         assertEquals("tensor<int8>(x[])", TensorType.fromSpec("tensor<int8>(x[])").toString());
     }
 
+    @Test
+    public void testIndexedSubtype() {
+        assertEquals(TensorType.fromSpec("tensor(x[10])"),
+                     TensorType.fromSpec("tensor(x[10])").indexedSubtype());
+        assertEquals(TensorType.fromSpec("tensor(x[10])"),
+                     TensorType.fromSpec("tensor(x[10],a{})").indexedSubtype());
+        assertEquals(TensorType.fromSpec("tensor(x[10],y[5])"),
+                     TensorType.fromSpec("tensor(x[10],y[5],a{},b{})").indexedSubtype());
+        assertEquals(TensorType.fromSpec("tensor()"),
+                     TensorType.fromSpec("tensor(a{})").indexedSubtype());
+    }
+
+    @Test
+    public void testMappedSubtype() {
+        assertEquals(TensorType.fromSpec("tensor(a{})"),
+                     TensorType.fromSpec("tensor(a{})").mappedSubtype());
+        assertEquals(TensorType.fromSpec("tensor(a{})"),
+                     TensorType.fromSpec("tensor(x[10],a{})").mappedSubtype());
+        assertEquals(TensorType.fromSpec("tensor(a{},b{})"),
+                     TensorType.fromSpec("tensor(x[10],y[5],a{},b{})").mappedSubtype());
+        assertEquals(TensorType.fromSpec("tensor()"),
+                     TensorType.fromSpec("tensor(x[10])").mappedSubtype());
+    }
+
     private static void assertTensorType(String typeSpec) {
         assertTensorType(typeSpec, typeSpec);
     }