Skip to content

Commit 356dd94

Browse files
authored
feat: Add support for Spark Pi math expression (#3789)
1 parent 1b4b26f commit 356dd94

2 files changed

Lines changed: 32 additions & 0 deletions

File tree

spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ object QueryPlanSerde extends Logging with CometExprShim with CometTypeShim {
114114
classOf[Log10] -> CometLog10,
115115
classOf[Logarithm] -> CometLogarithm,
116116
classOf[Multiply] -> CometMultiply,
117+
classOf[Pi] -> CometScalarFunction("pi"),
117118
classOf[Pow] -> CometScalarFunction("pow"),
118119
classOf[Rand] -> CometRand,
119120
classOf[Randn] -> CometRandn,
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
-- Licensed to the Apache Software Foundation (ASF) under one
2+
-- or more contributor license agreements. See the NOTICE file
3+
-- distributed with this work for additional information
4+
-- regarding copyright ownership. The ASF licenses this file
5+
-- to you under the Apache License, Version 2.0 (the
6+
-- "License"); you may not use this file except in compliance
7+
-- with the License. You may obtain a copy of the License at
8+
--
9+
-- http://www.apache.org/licenses/LICENSE-2.0
10+
--
11+
-- Unless required by applicable law or agreed to in writing,
12+
-- software distributed under the License is distributed on an
13+
-- "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14+
-- KIND, either express or implied. See the License for the
15+
-- specific language governing permissions and limitations
16+
-- under the License.
17+
18+
-- ConfigMatrix: parquet.enable.dictionary=false,true
19+
20+
statement
21+
CREATE TABLE test_pi(d double) USING parquet
22+
23+
statement
24+
INSERT INTO test_pi VALUES (1.0)
25+
26+
query tolerance=1e-15
27+
SELECT pi() FROM test_pi
28+
29+
-- literal arguments
30+
query tolerance=1e-15
31+
SELECT pi()

0 commit comments

Comments
 (0)