Add to data space delimiting test #147

BoiseState-AdaptLab · Oct 13, 2021 · b76267a · b76267a
1 parent 1317173
commit b76267a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/computation/computation_test.cc b/src/computation/computation_test.cc
@@ -1544,6 +1544,7 @@ TEST_F(ComputationTest, DelimitDataSpacesTest) {
     comp->addDataSpace("sd", "int");
     comp->addDataSpace("x", "int");
     comp->addDataSpace("z", "string");
+    comp->addDataSpace("N", "int");
 
     EXPECT_EQ("int $asdf$ = 3;", comp->delimitDataSpacesInString("int asdf = 3;"));
     EXPECT_EQ("int $asdf$ = 3*$x$;", comp->delimitDataSpacesInString("int asdf = 3*x;"));
@@ -1556,6 +1557,8 @@ TEST_F(ComputationTest, DelimitDataSpacesTest) {
 }", comp->delimitDataSpacesInString("if (z == \"\\\"asdf\") {\
     asdf = 5;\
 }"));
+    EXPECT_EQ("{[i,j]: 0 < i < j && 0 < j < 10}", comp->delimitDataSpacesInString("{[i,j]: 0 < i < j && 0 < j < 10}"));
+    EXPECT_EQ("{[i]: 0 < i < $N$}", comp->delimitDataSpacesInString("{[i]: 0 < i < N}"));
 
     delete comp;
 }