Skip to content

Commit f5fa616

Browse files
address review comments
1 parent 2a431fd commit f5fa616

2 files changed

Lines changed: 35 additions & 37 deletions

File tree

spark/src/test/resources/sql-tests/expressions/math/tan.sql

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,11 @@ statement
2222
CREATE TABLE test_tan(d double) USING parquet
2323

2424
statement
25-
INSERT INTO test_tan VALUES (0.0), (0.7853981633974483), (-0.7853981633974483), (1.0), (NULL), (cast('NaN' as double)), (cast('Infinity' as double))
25+
INSERT INTO test_tan VALUES (0.0), (-0.0), (0.7853981633974483), (-0.7853981633974483), (1.0), (NULL), (cast('NaN' as double)), (cast('Infinity' as double))
2626

2727
query tolerance=1e-6
2828
SELECT tan(d) FROM test_tan
2929

3030
-- literal arguments
3131
query tolerance=1e-6
32-
SELECT tan(0.0), tan(0.7853981633974483), tan(NULL)
32+
SELECT tan(0.0), tan(-0.0), tan(0.7853981633974483), tan(NULL)

spark/src/test/scala/org/apache/comet/CometExpressionSuite.scala

Lines changed: 33 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -1345,41 +1345,39 @@ class CometExpressionSuite extends CometTestBase with AdaptiveSparkPlanHelper {
13451345

13461346
test("various math scalar functions") {
13471347
val data = doubleValues.map(n => (n, n))
1348-
withSQLConf() {
1349-
withParquetTable(data, "tbl") {
1350-
// expressions with single arg
1351-
for (expr <- Seq(
1352-
"acos",
1353-
"asin",
1354-
"atan",
1355-
"cos",
1356-
"cosh",
1357-
"exp",
1358-
"ln",
1359-
"log10",
1360-
"log2",
1361-
"sin",
1362-
"sinh",
1363-
"sqrt",
1364-
"tan",
1365-
"tanh",
1366-
"cot")) {
1367-
val (_, cometPlan) =
1368-
checkSparkAnswerAndOperatorWithTol(sql(s"SELECT $expr(_1), $expr(_2) FROM tbl"))
1369-
val cometProjectExecs = collect(cometPlan) { case op: CometProjectExec =>
1370-
op
1371-
}
1372-
assert(cometProjectExecs.length == 1, expr)
1373-
}
1374-
// expressions with two args
1375-
for (expr <- Seq("pow")) {
1376-
val (_, cometPlan) =
1377-
checkSparkAnswerAndOperatorWithTol(sql(s"SELECT $expr(_1, _2) FROM tbl"))
1378-
val cometProjectExecs = collect(cometPlan) { case op: CometProjectExec =>
1379-
op
1380-
}
1381-
assert(cometProjectExecs.length == 1, expr)
1382-
}
1348+
withParquetTable(data, "tbl") {
1349+
// expressions with single arg
1350+
for (expr <- Seq(
1351+
"acos",
1352+
"asin",
1353+
"atan",
1354+
"cos",
1355+
"cosh",
1356+
"exp",
1357+
"ln",
1358+
"log10",
1359+
"log2",
1360+
"sin",
1361+
"sinh",
1362+
"sqrt",
1363+
"tan",
1364+
"tanh",
1365+
"cot")) {
1366+
val (_, cometPlan) =
1367+
checkSparkAnswerAndOperatorWithTol(sql(s"SELECT $expr(_1), $expr(_2) FROM tbl"))
1368+
val cometProjectExecs = collect(cometPlan) { case op: CometProjectExec =>
1369+
op
1370+
}
1371+
assert(cometProjectExecs.length == 1, expr)
1372+
}
1373+
// expressions with two args
1374+
for (expr <- Seq("pow")) {
1375+
val (_, cometPlan) =
1376+
checkSparkAnswerAndOperatorWithTol(sql(s"SELECT $expr(_1, _2) FROM tbl"))
1377+
val cometProjectExecs = collect(cometPlan) { case op: CometProjectExec =>
1378+
op
1379+
}
1380+
assert(cometProjectExecs.length == 1, expr)
13831381
}
13841382
}
13851383
}

0 commit comments

Comments
 (0)