Upgrade runner to Ubuntu 24.04 (#3445)

kuzudb · May 6, 2024 · 349f906 · 349f906
1 parent cc43956
commit 349f906
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/test/include/test_runner/test_group.h b/test/include/test_runner/test_group.h
@@ -36,6 +36,7 @@ struct TestStatement {
     bool enumerate = false;
     bool checkOutputOrder = false;
     bool checkColumnNames = false;
+    bool checkNumeric = false;
     std::vector<TestQueryResult> result;
     // for multiple conns
     std::string batchStatmentsCSVFile;

diff --git a/test/include/test_runner/test_parser.h b/test/include/test_runner/test_parser.h
@@ -18,6 +18,7 @@ enum class TokenType {
     BUFFER_POOL_SIZE,
     CASE,
     CHECK_COLUMN_NAMES,
+    CHECK_NUMERIC,
     CHECK_ORDER,
     COMMIT,
     DEFINE,
@@ -61,7 +62,7 @@ const std::unordered_map<std::string, TokenType> tokenMap = {{"-DATASET", TokenT
     {"-CREATE_CONNECTION", TokenType::CREATE_CONNECTION}, {"]", TokenType::END_OF_STATEMENT_BLOCK},
     {"----", TokenType::RESULT}, {"--", TokenType::SEPARATOR}, {"#", TokenType::EMPTY},
     {"-SET", TokenType::SET}, {"-IMPORT_DATABASE", TokenType::IMPORT_DATABASE},
-    {"-REMOVE_FILE", TokenType::REMOVE_FILE},
+    {"-REMOVE_FILE", TokenType::REMOVE_FILE}, {"-CHECK_NUMERIC", TokenType::CHECK_NUMERIC},
     {"-CHECK_COLUMN_NAMES", TokenType::CHECK_COLUMN_NAMES}};
 
 class LogicToken {

diff --git a/test/test_runner/test_parser.cpp b/test/test_runner/test_parser.cpp
@@ -272,6 +272,10 @@ TestStatement* TestParser::extractStatement(TestStatement* statement,
         statement->checkOutputOrder = true;
         break;
     }
+    case TokenType::CHECK_NUMERIC: {
+        statement->checkNumeric = true;
+        break;
+    }
     case TokenType::ENUMERATE: {
         statement->enumerate = true;
         break;