join dimensions must equal common dimensions for optimization

1 files changed, 40 insertions, 0 deletions

diff --git a/vespajlib/src/test/java/com/yahoo/tensor/functions/ReduceJoinTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/functions/ReduceJoinTestCase.java
new file mode 100644
index 00000000000..d073c60d993
--- /dev/null
+++ b/vespajlib/src/test/java/com/yahoo/tensor/functions/ReduceJoinTestCase.java
@@ -0,0 +1,40 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.tensor.functions;
+
+import com.yahoo.tensor.Tensor;
+import com.yahoo.tensor.evaluation.Name;
+import org.junit.Test;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author arnej
+ */
+public class ReduceJoinTestCase {
+
+    @Test
+    public void testReduceJoinTriggers() {
+        var a = Tensor.from("tensor(x[3])", "[1,2,3]");
+        var b = Tensor.from("tensor(x[3])", "[4,5,6]");
+        var fa = new ConstantTensor<Name>(a);
+        var fb = new ConstantTensor<Name>(a);
+        var j = new Join<Name>(fa, fb, ScalarFunctions.add());
+        var r = new Reduce<Name>(j, Reduce.Aggregator.sum, "x");
+        var rj = new ReduceJoin<Name>(r, j);
+        assertTrue(rj.canOptimize(a, b));
+    }
+
+    @Test
+    public void testReduceJoinUnoptimized() {
+        var a = Tensor.from("tensor(x[3])", "[1,2,3]");
+        var b = Tensor.from("tensor(y[3])", "[4,5,6]");
+        var fa = new ConstantTensor<Name>(a);
+        var fb = new ConstantTensor<Name>(a);
+        var j = new Join<Name>(fa, fb, ScalarFunctions.add());
+        var r = new Reduce<Name>(j, Reduce.Aggregator.sum, "x");
+        var rj = new ReduceJoin<Name>(r, j);
+        assertFalse(rj.canOptimize(a, b));
+    }
+}