1 files changed, 89 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
new file mode 100644
index 00000000000..59d77f6569a
--- /dev/null
+++ b/vespajlib/src/test/java/com/yahoo/tensor/TensorTypeTestCase.java
@@ -0,0 +1,89 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.tensor;
+
+import org.junit.Test;
+
+import static org.hamcrest.Matchers.containsString;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.fail;
+
+/**
+ * @author <a href="mailto:geirst@yahoo-inc.com">Geir Storli</a>
+ */
+public class TensorTypeTestCase {
+
+    @Test
+    public void requireThatAnEmptyTensorTypeCanBeSpecified() {
+        assertTensorType("tensor()");
+    }
+
+    @Test
+    public void requireThatBoundIndexedDimensionsCanBeSpecified() {
+        assertTensorType("tensor(x[5])");
+        assertTensorType("tensor(x[5],y[10],z[100])");
+        assertTensorType("tensor(x[5],y[10],z[100])", "tensor( x[5] , y[10] , z[100] )");
+        assertTensorType("tensor(baR_09[10])");
+    }
+
+    @Test
+    public void requireThatUnboundIndexedDimensionsCanBeSpecified() {
+        assertTensorType("tensor(x[])");
+        assertTensorType("tensor(x[],y[],z[])");
+        assertTensorType("tensor(x[],y[],z[])", "tensor( x[] , y[] , z[] )");
+        assertTensorType("tensor(baR_09[])");
+    }
+
+    @Test
+    public void requireThatMappedDimensionsCanBeSpecified() {
+        assertTensorType("tensor(x{})");
+        assertTensorType("tensor(x{},y{},z{})");
+        assertTensorType("tensor(x{},y{},z{})", "tensor( x{} , y{} , z{} )");
+        assertTensorType("tensor(baR_09{})");
+    }
+
+    @Test
+    public void requireThatIndexedBoundDimensionMustHaveNonZeroSize() {
+        assertIllegalTensorType("tensor(x[0])", "Size of bound dimension 'x' must be at least 1");
+    }
+
+    @Test
+    public void requireThatDimensionsMustHaveUniqueNames() {
+        assertIllegalTensorType("tensor(x[10],y[20],x[30])", "'x[10]' and 'x[30]' have the same name");
+        assertIllegalTensorType("tensor(x{},y{},x{})", "'x{}' and 'x{}' have the same name");
+    }
+
+    @Test
+    public void requireThatDimensionsAreOfSameType() {
+        assertIllegalTensorType("tensor(x[10],y[])", "'x[10]' does not have the same type as 'y[]'");
+        assertIllegalTensorType("tensor(x[10],y{})", "'x[10]' does not have the same type as 'y{}'");
+        assertIllegalTensorType("tensor(x[10],y[20],z{})", "'y[20]' does not have the same type as 'z{}'");
+        assertIllegalTensorType("tensor(x[],y{})", "'x[]' does not have the same type as 'y{}'");
+    }
+
+    @Test
+    public void requireThatIllegalSyntaxInSpecThrowsException() {
+        assertIllegalTensorType("foo(x[10])", "Tensor type spec must start with 'tensor(' and end with ')', but was 'foo(x[10])'");
+        assertIllegalTensorType("tensor(x_@[10])", "Failed parsing element 'x_@[10]' in type spec 'tensor(x_@[10])'");
+        assertIllegalTensorType("tensor(x[10a])", "Failed parsing element 'x[10a]' in type spec 'tensor(x[10a])'");
+        assertIllegalTensorType("tensor(x{10})", "Failed parsing element 'x{10}' in type spec 'tensor(x{10})'");
+    }
+
+    private static void assertTensorType(String typeSpec) {
+        assertTensorType(typeSpec, typeSpec);
+    }
+
+    private static void assertTensorType(String expected, String typeSpec) {
+        assertEquals(expected, TensorType.fromSpec(typeSpec).toString());
+    }
+
+    private static void assertIllegalTensorType(String typeSpec, String messageSubstring) {
+        try {
+            TensorType.fromSpec(typeSpec);
+            fail("Exception exception to be thrown with message: '" + messageSubstring + "'");
+        } catch (IllegalArgumentException e) {
+            assertThat(e.getMessage(), containsString(messageSubstring));
+        }
+    }
+
+}