Support float type for sum function (#776)

native-engine/datafusion-ext-plans/src/agg/agg.rs

@@ -176,6 +176,7 @@ pub fn create_agg(
                 DataType::Int8 | DataType::Int16 | DataType::Int32 | DataType::Int64 => {
                     DataType::Int64
                 }
+                DataType::Float32 | DataType::Float64 => DataType::Float64,
                 other => sum_return_type(&other)?,
             };
             Arc::new(sum::AggSum::try_new(

spark-extension-shims-spark3/src/test/scala/org/apache/spark/sql/blaze/BlazeFunctionSuite.scala

@@ -0,0 +1,33 @@
+/*
+ * Copyright 2022 The Blaze Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.apache.spark.sql.blaze
+
+import org.apache.spark.sql.Row
+
+class BlazeFunctionSuite extends org.apache.spark.sql.QueryTest with BaseBlazeSQLSuite {
+
+  test("sum function with float input") {
+    withTable("t1") {
+      withSQLConf("spark.blaze.enable" -> "false") {
+      sql("set spark.blaze.enable=false")
+      sql("create table t1 using parquet as select 1.0f as c1")
+      val df = sql("select sum(c1) from t1")
+      checkAnswer(df, Seq(Row(1.23, 1.1)))
+      }
+    }
+  }
+
+}