summaryrefslogtreecommitdiffstats
path: root/eval/src
diff options
context:
space:
mode:
Diffstat (limited to 'eval/src')
-rw-r--r--eval/src/tests/eval/value_type/value_type_test.cpp14
1 files changed, 14 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 d0b48759578..85ff7613775 100644
--- a/eval/src/tests/eval/value_type/value_type_test.cpp
+++ b/eval/src/tests/eval/value_type/value_type_test.cpp
@@ -301,6 +301,20 @@ TEST("require that type-related predicate functions work as expected") {
TEST_DO(verify_predicates(type("tensor<float>(x[5],y{})"), false, false, true, false, false));
}
+TEST("require that dense subspace size calculation works as expected") {
+ EXPECT_EQUAL(type("error").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("double").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("tensor()").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("tensor(x{})").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("tensor(x{},y{})").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("tensor(x[5])").dense_subspace_size(), 5u);
+ EXPECT_EQUAL(type("tensor(x[5],y[10])").dense_subspace_size(), 50u);
+ EXPECT_EQUAL(type("tensor(x[5],y{})").dense_subspace_size(), 5u);
+ EXPECT_EQUAL(type("tensor<float>(x{})").dense_subspace_size(), 1u);
+ EXPECT_EQUAL(type("tensor<float>(x[5])").dense_subspace_size(), 5u);
+ EXPECT_EQUAL(type("tensor<float>(x[5],y{})").dense_subspace_size(), 5u);
+}
+
TEST("require that dimension predicates work as expected") {
ValueType::Dimension x("x");
ValueType::Dimension y("y", 10);