diff --git a/presto-native-execution/presto_cpp/main/types/tests/PrestoToVeloxQueryPlanTest.cpp b/presto-native-execution/presto_cpp/main/types/tests/PrestoToVeloxQueryPlanTest.cpp index 1457805745ea5..9f22ff7ef0e5d 100644 --- a/presto-native-execution/presto_cpp/main/types/tests/PrestoToVeloxQueryPlanTest.cpp +++ b/presto-native-execution/presto_cpp/main/types/tests/PrestoToVeloxQueryPlanTest.cpp @@ -19,6 +19,7 @@ #include "velox/functions/prestosql/types/IPAddressType.h" #include "velox/functions/prestosql/types/IPPrefixType.h" #include "velox/functions/prestosql/types/JsonType.h" +#include "velox/functions/prestosql/types/TDigestType.h" #include "velox/functions/prestosql/types/TimestampWithTimeZoneType.h" #include "velox/functions/prestosql/types/UuidType.h" @@ -50,6 +51,7 @@ class PrestoToVeloxQueryPlanTest : public ::testing::Test { registerIPAddressType(); registerIPPrefixType(); registerJsonType(); + registerTDigestType(); registerTimestampWithTimeZoneType(); registerUuidType(); } @@ -111,7 +113,7 @@ TEST_F(PrestoToVeloxQueryPlanTest, parseSqlFunctionHandleAllComplexTypes) { std::string str = R"( { "@type": "json_file", - "functionId": "json_file.test.all_complex_types;row(map(hugeint, ipaddress), ipprefix);row(array(varbinary), timestamp, date, json, hyperloglog, timestamp with time zone, interval year to month, interval day to second);function(double, boolean);uuid", + "functionId": "json_file.test.all_complex_types;row(map(hugeint, ipaddress), ipprefix);row(array(varbinary), timestamp, date, json, hyperloglog, tdigest, timestamp with time zone, interval year to month, interval day to second);function(double, boolean);uuid", "version": "1" } )"; @@ -129,6 +131,7 @@ TEST_F(PrestoToVeloxQueryPlanTest, parseSqlFunctionHandleAllComplexTypes) { DATE(), JSON(), HYPERLOGLOG(), + TDIGEST(), TIMESTAMP_WITH_TIME_ZONE(), INTERVAL_YEAR_MONTH(), INTERVAL_DAY_TIME()}),