This is an automated email from the ASF dual-hosted git repository.
jonah pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion.git
The following commit(s) were added to refs/heads/main by this push:
new a534e853e8 Minor: Add tests for types of arithmetic operator output
types (#14250)
a534e853e8 is described below
commit a534e853e88365b67363b26a004f6037c2f3a5b0
Author: Andrew Lamb <and...@nerdnetworks.org>
AuthorDate: Thu Jan 23 21:01:20 2025 -0500
Minor: Add tests for types of arithmetic operator output types (#14250)
---
datafusion/sqllogictest/test_files/operator.slt | 114 ++++++++++++++++++++++++
1 file changed, 114 insertions(+)
diff --git a/datafusion/sqllogictest/test_files/operator.slt
b/datafusion/sqllogictest/test_files/operator.slt
new file mode 100644
index 0000000000..ac977a0c51
--- /dev/null
+++ b/datafusion/sqllogictest/test_files/operator.slt
@@ -0,0 +1,114 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you 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.
+
+# Validate arithmetic operator output type
+
+statement ok
+create table numeric_types as
+SELECT
+ arrow_cast(1, 'Int8') as int8,
+ arrow_cast(1, 'Int16') as int16,
+ arrow_cast(1, 'Int32') as int32,
+ arrow_cast(1, 'Int64') as int64,
+ arrow_cast(1, 'UInt8') as uint8,
+ arrow_cast(1, 'UInt16') as uint16,
+ arrow_cast(1, 'UInt32') as uint32,
+ arrow_cast(1, 'UInt64') as uint64,
+ arrow_cast(1.23, 'Float32') as float32,
+ arrow_cast(1.23, 'Float64') as float64,
+ arrow_cast(1.25, 'Decimal128(5, 2)') as decimal
+;
+
+# Plus with the same operand type, expect the same output type
+# except for decimal which is promoted to the highest precision
+query TTTTTTTTTTT
+select
+ arrow_typeof(int8 + int8),
+ arrow_typeof(int16 + int16),
+ arrow_typeof(int32 + int32),
+ arrow_typeof(int64 + int64),
+ arrow_typeof(uint8 + uint8),
+ arrow_typeof(uint16 + uint16),
+ arrow_typeof(uint32 + uint32),
+ arrow_typeof(uint64 + uint64),
+ arrow_typeof(float32 + float32),
+ arrow_typeof(float64 + float64),
+ arrow_typeof(decimal + decimal)
+from numeric_types;
+----
+Int8 Int16 Int32 Int64 UInt8 UInt16 UInt32 UInt64 Float32 Float64
Decimal128(6, 2)
+
+
+# Minus with the same operand type, expect the same output type
+# except for decimal which is promoted to the highest precision
+query TTTTTTTTTTT
+select
+ arrow_typeof(int8 - int8),
+ arrow_typeof(int16 - int16),
+ arrow_typeof(int32 - int32),
+ arrow_typeof(int64 - int64),
+ arrow_typeof(uint8 - uint8),
+ arrow_typeof(uint16 - uint16),
+ arrow_typeof(uint32 - uint32),
+ arrow_typeof(uint64 - uint64),
+ arrow_typeof(float32 - float32),
+ arrow_typeof(float64 - float64),
+ arrow_typeof(decimal - decimal)
+from numeric_types;
+----
+Int8 Int16 Int32 Int64 UInt8 UInt16 UInt32 UInt64 Float32 Float64
Decimal128(6, 2)
+
+# Multiply with the same operand type, expect the same output type
+# except for decimal which is promoted to the highest precision
+query TTTTTTTTTTT
+select
+ arrow_typeof(int8 * int8),
+ arrow_typeof(int16 * int16),
+ arrow_typeof(int32 * int32),
+ arrow_typeof(int64 * int64),
+ arrow_typeof(uint8 * uint8),
+ arrow_typeof(uint16 * uint16),
+ arrow_typeof(uint32 * uint32),
+ arrow_typeof(uint64 * uint64),
+ arrow_typeof(float32 * float32),
+ arrow_typeof(float64 * float64),
+ arrow_typeof(decimal * decimal)
+from numeric_types;
+----
+Int8 Int16 Int32 Int64 UInt8 UInt16 UInt32 UInt64 Float32 Float64
Decimal128(11, 4)
+
+# Divide with the same operand type, expect the same output type
+# except for decimal which is promoted to the highest precision
+query TTTTTTTTTTT
+select
+ arrow_typeof(int8 / int8),
+ arrow_typeof(int16 / int16),
+ arrow_typeof(int32 / int32),
+ arrow_typeof(int64 / int64),
+ arrow_typeof(uint8 / uint8),
+ arrow_typeof(uint16 / uint16),
+ arrow_typeof(uint32 / uint32),
+ arrow_typeof(uint64 / uint64),
+ arrow_typeof(float32 / float32),
+ arrow_typeof(float64 / float64),
+ arrow_typeof(decimal / decimal)
+from numeric_types;
+----
+Int8 Int16 Int32 Int64 UInt8 UInt16 UInt32 UInt64 Float32 Float64
Decimal128(11, 6)
+
+statement ok
+drop table numeric_types
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@datafusion.apache.org
For additional commands, e-mail: commits-h...@datafusion.apache.org
