1 files changed, 24 insertions, 0 deletions
diff --git a/eval/src/tests/eval/value_cache/tensor_loader_test.cpp b/eval/src/tests/eval/value_cache/tensor_loader_test.cpp
index 1a77cfe847b..4b4ba3fc0d3 100644
--- a/eval/src/tests/eval/value_cache/tensor_loader_test.cpp
+++ b/eval/src/tests/eval/value_cache/tensor_loader_test.cpp
@@ -19,12 +19,26 @@ TensorSpec make_dense_tensor() {
         .add({{"x", 1}, {"y", 1}}, 4.0);
 }
 
+TensorSpec make_simple_dense_tensor() {
+    return TensorSpec("tensor(z[3])")
+        .add({{"z", 0}}, 1.0)
+        .add({{"z", 1}}, 2.0)
+        .add({{"z", 2}}, 3.5);
+}
+
 TensorSpec make_sparse_tensor() {
     return TensorSpec("tensor(x{},y{})")
         .add({{"x", "foo"}, {"y", "bar"}}, 1.0)
         .add({{"x", "bar"}, {"y", "foo"}}, 2.0);
 }
 
+TensorSpec make_simple_sparse_tensor() {
+    return TensorSpec("tensor(mydim{})")
+        .add({{"mydim", "foo"}}, 1.0)
+        .add({{"mydim", "three"}}, 3.0)
+        .add({{"mydim", "bar"}}, 2.0);
+}
+
 TensorSpec make_mixed_tensor() {
     return TensorSpec("tensor(x{},y[2])")
         .add({{"x", "foo"}, {"y", 0}}, 1.0)
@@ -75,6 +89,16 @@ TEST_F("require that lz4 compressed sparse tensor can be loaded", ConstantTensor
     TEST_DO(verify_tensor(make_sparse_tensor(), f1.create(TEST_PATH("sparse.json.lz4"), "tensor(x{},y{})")));
 }
 
+TEST_F("require that sparse tensor short form can be loaded", ConstantTensorLoader(factory)) {
+    TEST_DO(verify_tensor(make_simple_sparse_tensor(), f1.create(TEST_PATH("sparse-short1.json"), "tensor(mydim{})")));
+    TEST_DO(verify_tensor(make_simple_sparse_tensor(), f1.create(TEST_PATH("sparse-short2.json"), "tensor(mydim{})")));
+}
+
+TEST_F("require that dense tensor short form can be loaded", ConstantTensorLoader(factory)) {
+    TEST_DO(verify_tensor(make_simple_dense_tensor(), f1.create(TEST_PATH("dense-short1.json"), "tensor(z[3])")));
+    TEST_DO(verify_tensor(make_simple_dense_tensor(), f1.create(TEST_PATH("dense-short2.json"), "tensor(z[3])")));
+}
+
 TEST_F("require that bad lz4 file fails to load creating empty result", ConstantTensorLoader(factory)) {
     TEST_DO(verify_tensor(sparse_tensor_nocells(), f1.create(TEST_PATH("bad_lz4.json.lz4"), "tensor(x{},y{})")));
 }