fix integration tests

zhtaoxiang · zhtaoxiang · commit 9df6837305f5 · 2023-09-05T23:37:59.000-07:00
diff --git a/pinot-query-planner/src/main/java/org/apache/pinot/query/type/TypeFactory.java b/pinot-query-planner/src/main/java/org/apache/pinot/query/type/TypeFactory.java
@@ -71,6 +71,8 @@ private RelDataType toRelDataType(FieldSpec fieldSpec) {
       // With float and double mapped to the same RelDataType, the behavior in multi-stage query engine will be the same
       // as the query in v1 query engine.
       case FLOAT:
+        return fieldSpec.isSingleValueField() ? createSqlType(SqlTypeName.DOUBLE)
+            : createArrayType(createSqlType(SqlTypeName.REAL), -1);
       case DOUBLE:
         return fieldSpec.isSingleValueField() ? createSqlType(SqlTypeName.DOUBLE)
             : createArrayType(createSqlType(SqlTypeName.DOUBLE), -1);
diff --git a/pinot-query-planner/src/test/java/org/apache/pinot/query/type/TypeFactoryTest.java b/pinot-query-planner/src/test/java/org/apache/pinot/query/type/TypeFactoryTest.java
@@ -87,6 +87,9 @@ public void testRelDataTypeConversion() {
               new ArraySqlType(new BasicSqlType(TYPE_SYSTEM, SqlTypeName.BIGINT), false));
           break;
         case "FLOAT_ARRAY_COL":
+          Assert.assertEquals(field.getType(),
+              new ArraySqlType(new BasicSqlType(TYPE_SYSTEM, SqlTypeName.REAL), false));
+          break;
         case "DOUBLE_ARRAY_COL":
           Assert.assertEquals(field.getType(),
               new ArraySqlType(new BasicSqlType(TYPE_SYSTEM, SqlTypeName.DOUBLE), false));
