Fix unit test

dremio · Nov 9, 2023 · fb55515 · fb55515
1 parent 7c7a939
commit fb55515
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/java/gandiva/src/test/java/org/apache/arrow/gandiva/evaluator/ExpressionRegistryTest.java b/java/gandiva/src/test/java/org/apache/arrow/gandiva/evaluator/ExpressionRegistryTest.java
@@ -39,7 +39,7 @@ public void testTypes() throws GandivaException {
   public void testFunctions() throws GandivaException {
     ArrowType.Int uint8 = new ArrowType.Int(8, false);
     FunctionSignature signature =
-        new FunctionSignature("add", uint8, null, Lists.newArrayList(uint8, uint8));
+        new FunctionSignature("add", uint8, ArrowType.Null(), Lists.newArrayList(uint8, uint8));
     Set<FunctionSignature> functions = ExpressionRegistry.getInstance().getSupportedFunctions();
     Assert.assertTrue(functions.contains(signature));
   }
@@ -48,7 +48,7 @@ public void testFunctions() throws GandivaException {
   public void testFunctionAliases() throws GandivaException {
     ArrowType.Int int64 = new ArrowType.Int(64, true);
     FunctionSignature signature =
-        new FunctionSignature("modulo", int64, null, Lists.newArrayList(int64, int64));
+        new FunctionSignature("modulo", int64, ArrowType.Null(), Lists.newArrayList(int64, int64));
     Set<FunctionSignature> functions = ExpressionRegistry.getInstance().getSupportedFunctions();
     Assert.assertTrue(functions.contains(signature));
   }
@@ -58,7 +58,7 @@ public void testCaseInsensitiveFunctionName() throws GandivaException {
     ArrowType.Utf8 utf8 = new ArrowType.Utf8();
     ArrowType.Int int64 = new ArrowType.Int(64, true);
     FunctionSignature signature =
-        new FunctionSignature("castvarchar", utf8, null, Lists.newArrayList(utf8, int64));
+        new FunctionSignature("castvarchar", utf8, ArrowType.Null(), Lists.newArrayList(utf8, int64));
     Set<FunctionSignature> functions = ExpressionRegistry.getInstance().getSupportedFunctions();
     Assert.assertTrue(functions.contains(signature));
   }