diff options
author | Lester Solbakken <lesters@oath.com> | 2021-04-22 09:21:05 +0200 |
---|---|---|
committer | Lester Solbakken <lesters@oath.com> | 2021-04-22 09:21:05 +0200 |
commit | bd531088df9eb9f7e083fd03d91e5d3f19a1664b (patch) | |
tree | 4c891ef1e0f41244fc150fabe955f163425e2bcc /vespajlib/src/test/java/com/yahoo | |
parent | 1f7363ff53144c40fd27c4332b1cb3619b1525d6 (diff) |
Wire in tensor cell type resolving for concat in Java
Diffstat (limited to 'vespajlib/src/test/java/com/yahoo')
-rw-r--r-- | vespajlib/src/test/java/com/yahoo/tensor/TypeResolverTestCase.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/vespajlib/src/test/java/com/yahoo/tensor/TypeResolverTestCase.java b/vespajlib/src/test/java/com/yahoo/tensor/TypeResolverTestCase.java index b7f998c6cf7..7eee50c6785 100644 --- a/vespajlib/src/test/java/com/yahoo/tensor/TypeResolverTestCase.java +++ b/vespajlib/src/test/java/com/yahoo/tensor/TypeResolverTestCase.java @@ -225,11 +225,14 @@ public class TypeResolverTestCase { checkConcat("tensor<float>(x[3])", "tensor()", "x", "tensor<float>(x[4])"); checkConcat("tensor<bfloat16>(x[3])", "tensor()", "x", "tensor<bfloat16>(x[4])"); checkConcat("tensor<int8>(x[3])", "tensor()", "x", "tensor<int8>(x[4])"); + // specific for Java + checkConcat("tensor(x[])", "tensor(x[2])", "x", "tensor(x[])"); + checkConcat("tensor(x[])", "tensor(x[2])", "y", "tensor(x[],y[2])"); + checkConcat("tensor(x[3])", "tensor(x[2])", "y", "tensor(x[2],y[2])"); // invalid combinations must fail checkConcatFails("tensor(x{})", "tensor(x[2])", "x"); checkConcatFails("tensor(x{})", "tensor(x{})", "x"); checkConcatFails("tensor(x{})", "tensor()", "x"); - checkConcatFails("tensor(x[3])", "tensor(x[2])", "y"); } @Test |