Testing changes

velox/functions/prestosql/tests/CMakeLists.txt

@@ -16,96 +16,9 @@ add_subdirectory(utils)
 
 add_executable(
   velox_functions_test
-  ArithmeticTest.cpp
-  ArrayAllMatchTest.cpp
-  ArrayAnyMatchTest.cpp
-  ArrayAverageTest.cpp
-  ArrayCombinationsTest.cpp
-  ArrayConcatTest.cpp
-  ArrayConstructorTest.cpp
-  ArrayContainsTest.cpp
-  ArrayCumSumTest.cpp
-  ArrayDistinctTest.cpp
-  ArrayDuplicatesTest.cpp
-  ArrayExceptTest.cpp
-  ArrayFilterTest.cpp
-  ArrayFrequencyTest.cpp
-  ArrayFlattenTest.cpp
-  ArrayHasDuplicatesTest.cpp
-  ArrayIntersectTest.cpp
-  ArrayJoinTest.cpp
-  ArrayMaxTest.cpp
-  ArrayMinTest.cpp
-  ArrayNGramsTest.cpp
-  ArrayNoneMatchTest.cpp
-  ArrayNormalizeTest.cpp
-  ArrayPositionTest.cpp
-  ArrayRemoveNullsTest.cpp
-  ArrayRemoveTest.cpp
-  ArrayShuffleTest.cpp
-  ArraySortTest.cpp
-  ArraysOverlapTest.cpp
-  ArraySumTest.cpp
-  ArrayTrimTest.cpp
-  ArrayUnionTest.cpp
-  ArgGeneratorTest.cpp
-  BinaryFunctionsTest.cpp
-  BitwiseTest.cpp
-  CardinalityTest.cpp
-  CeilFloorTest.cpp
-  ComparisonsTest.cpp
-  DateTimeFunctionsTest.cpp
-  DecimalArithmeticTest.cpp
-  ElementAtTest.cpp
-  FailTest.cpp
-  FindFirstTest.cpp
-  FromUtf8Test.cpp
-  GreatestLeastTest.cpp
-  HyperLogLogCastTest.cpp
-  HyperLogLogFunctionsTest.cpp
+
   IPAddressFunctionsTest.cpp
-  InPredicateTest.cpp
-  JsonCastTest.cpp
-  JsonExtractScalarTest.cpp
-  JsonFunctionsTest.cpp
-  MapEntriesTest.cpp
-  MapFilterTest.cpp
-  MapFromEntriesTest.cpp
-  MapNormalizeTest.cpp
-  MapTopNTest.cpp
-  MapTopNKeysTest.cpp
-  MapKeysAndValuesTest.cpp
-  MapMatchTest.cpp
-  MapTest.cpp
-  MapZipWithTest.cpp
-  MultimapFromEntriesTest.cpp
-  NotTest.cpp
-  ProbabilityTest.cpp
-  RandTest.cpp
-  ReduceTest.cpp
-  RegexpReplaceTest.cpp
-  ReverseTest.cpp
-  RoundTest.cpp
-  ScalarFunctionRegTest.cpp
-  SequenceTest.cpp
-  SimpleComparisonMatcherTest.cpp
-  SliceTest.cpp
-  SplitTest.cpp
-  SplitToMapTest.cpp
-  StringFunctionsTest.cpp
-  TimestampWithTimeZoneCastTest.cpp
-  TransformKeysTest.cpp
-  TransformTest.cpp
-  TransformValuesTest.cpp
-  TrimFunctionsTest.cpp
-  TypeOfTest.cpp
-  URLFunctionsTest.cpp
-  Utf8Test.cpp
-  UuidFunctionsTest.cpp
-  WidthBucketArrayTest.cpp
-  WordStemTest.cpp
-  ZipTest.cpp
-  ZipWithTest.cpp)
+  )
 
 add_test(velox_functions_test velox_functions_test)
 

velox/functions/prestosql/tests/IPAddressFunctionsTest.cpp

@@ -39,17 +39,17 @@ class IPAddressTest : public functions::test::FunctionBaseTest {
 
   std::optional<std::string> getIPPrefix(
       const std::optional<std::string> input,
-      std::optional<int8_t> mask) {
+      std::optional<int64_t> mask) {
     auto result = evaluateOnce<std::string>(
-        "cast(ip_prefix(cast(c0 as ipaddress), c1) as varchar)", input, mask);
+        "cast(ip_prefix(cast(c0 as ipaddress), c1) as varchar)", {VARCHAR(), BIGINT()}, input, mask);
     return result;
   }
 
   std::optional<std::string> getIPPrefixUsingVarchar(
       const std::optional<std::string> input,
-      std::optional<int8_t> mask) {
+      std::optional<int64_t> mask) {
     auto result = evaluateOnce<std::string>(
-        "cast(ip_prefix(c0, c1) as varchar)", input, mask);
+        "cast(ip_prefix(c0, c1) as varchar)", {VARCHAR(), BIGINT()}, input, mask);
     return result;
   }