diff options
| author | Henning Baldersheim <balder@yahoo-inc.com> | 2019-04-08 15:24:48 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2019-04-08 15:24:48 +0200 |
| commit | 6935cdc74fe60a94b26091413f46218c9833f0be (patch) | |
| tree | 25a37dfafbd7d61f83ad48111a36905e5cadc9dd /eval/src/tests | |
| parent | ceee063299f3a10b37975152a0859afd2d5072a0 (diff) | |
| parent | b84078e4bcb7bad171f42d40b7ee5337ded3ed35 (diff) | |
Merge pull request #9029 from vespa-engine/havardpe/parse-and-ignore-tensor-cell-type
parse (and ignore for now) tensor cell type in tensor type spec
Diffstat (limited to 'eval/src/tests')
| -rw-r--r-- | eval/src/tests/eval/value_type/value_type_test.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/eval/src/tests/eval/value_type/value_type_test.cpp b/eval/src/tests/eval/value_type/value_type_test.cpp index ffdc601932e..f7db7816fad 100644 --- a/eval/src/tests/eval/value_type/value_type_test.cpp +++ b/eval/src/tests/eval/value_type/value_type_test.cpp @@ -268,6 +268,8 @@ TEST("require that value type spec can be parsed") { EXPECT_EQUAL(ValueType::tensor_type({{"y", 10}}), ValueType::from_spec("tensor(y[10])")); EXPECT_EQUAL(ValueType::tensor_type({{"z", 0}}), ValueType::from_spec("tensor(z[])")); EXPECT_EQUAL(ValueType::tensor_type({{"x"}, {"y", 10}, {"z", 0}}), ValueType::from_spec("tensor(x{},y[10],z[])")); + EXPECT_EQUAL(ValueType::tensor_type({{"y", 10}}), ValueType::from_spec("tensor<double>(y[10])")); + EXPECT_EQUAL(ValueType::tensor_type({{"y", 10}}), ValueType::from_spec("tensor<float>(y[10])")); } TEST("require that value type spec can be parsed with extra whitespace") { @@ -280,6 +282,8 @@ TEST("require that value type spec can be parsed with extra whitespace") { EXPECT_EQUAL(ValueType::tensor_type({{"z", 0}}), ValueType::from_spec(" tensor ( z [ ] ) ")); EXPECT_EQUAL(ValueType::tensor_type({{"x"}, {"y", 10}, {"z", 0}}), ValueType::from_spec(" tensor ( x { } , y [ 10 ] , z [ ] ) ")); + EXPECT_EQUAL(ValueType::tensor_type({{"y", 10}}), ValueType::from_spec(" tensor < double > ( y [ 10 ] ) ")); + EXPECT_EQUAL(ValueType::tensor_type({{"y", 10}}), ValueType::from_spec(" tensor < float > ( y [ 10 ] ) ")); } TEST("require that malformed value type spec is parsed as error") { @@ -300,6 +304,7 @@ TEST("require that malformed value type spec is parsed as error") { EXPECT_TRUE(ValueType::from_spec("tensor(x{},x{})").is_error()); EXPECT_TRUE(ValueType::from_spec("tensor(x{},x[10])").is_error()); EXPECT_TRUE(ValueType::from_spec("tensor(x{},x[])").is_error()); + EXPECT_TRUE(ValueType::from_spec("tensor<float16>(x[10])").is_error()); } struct ParseResult { |