thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-24 08:32:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

[chore] add test for arrayCumSumNonNegative.

Browse Source 
Signed-off-by: clundro <859287553@qq.com>
This commit is contained in:
clundro 2023-04-18 01:36:38 +08:00 committed by cluster
parent a2c8661936
commit f5e07f84ff
3 changed files with 54 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/Functions/array/arrayCumSumNonNegative.cpp
View File

@ -30,11 +30,25 @@ struct ArrayCumSumNonNegativeImpl
{ {
WhichDataType which(expression_return); WhichDataType which(expression_return);
if (which.isUInt())
{
if (which.isNativeUInt()) if (which.isNativeUInt())
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeUInt64>()); return std::make_shared<DataTypeArray>(std::make_shared<DataTypeUInt64>());
if (which.isUInt256())
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeUInt256>());
else
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeUInt128>());
}
if (which.isInt())
{
if (which.isNativeInt()) if (which.isNativeInt())
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeInt64>()); return std::make_shared<DataTypeArray>(std::make_shared<DataTypeInt64>());
if (which.isInt256())
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeInt256>());
else
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeInt128>());
}
if (which.isFloat()) if (which.isFloat())
return std::make_shared<DataTypeArray>(std::make_shared<DataTypeFloat64>()); return std::make_shared<DataTypeArray>(std::make_shared<DataTypeFloat64>());
@ -50,14 +64,14 @@ struct ArrayCumSumNonNegativeImpl
return std::make_shared<DataTypeArray>(nested); return std::make_shared<DataTypeArray>(nested);
} }
throw Exception(ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT, throw Exception(
"arrayCumSumNonNegativeImpl cannot add values of type {}", expression_return->getName()); ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT, "arrayCumSumNonNegativeImpl cannot add values of type {}", expression_return->getName());
} }
template <typename Src, typename Dst> template <typename Src, typename Dst>
static void NO_SANITIZE_UNDEFINED implVector( static void NO_SANITIZE_UNDEFINED
size_t size, const IColumn::Offset * __restrict offsets, Dst * __restrict res_values, const Src * __restrict src_values) implVector(size_t size, const IColumn::Offset * __restrict offsets, Dst * __restrict res_values, const Src * __restrict src_values)
{ {
size_t pos = 0; size_t pos = 0;
for (const auto * end = offsets + size; offsets < end; ++offsets) for (const auto * end = offsets + size; offsets < end; ++offsets)
@ -100,7 +114,6 @@ struct ArrayCumSumNonNegativeImpl
implVector(offsets.size(), offsets.data(), res_values.data(), data.data()); implVector(offsets.size(), offsets.data(), res_values.data(), data.data());
res_ptr = ColumnArray::create(std::move(res_nested), array.getOffsetsPtr()); res_ptr = ColumnArray::create(std::move(res_nested), array.getOffsetsPtr());
return true; return true;
} }
static ColumnPtr execute(const ColumnArray & array, ColumnPtr mapped) static ColumnPtr execute(const ColumnArray & array, ColumnPtr mapped)
@ -108,28 +121,25 @@ struct ArrayCumSumNonNegativeImpl
ColumnPtr res; ColumnPtr res;
mapped = mapped->convertToFullColumnIfConst(); mapped = mapped->convertToFullColumnIfConst();
if (executeType< UInt8 , UInt64>(mapped, array, res) || if (executeType<UInt8, UInt64>(mapped, array, res) || executeType<UInt16, UInt64>(mapped, array, res)
executeType< UInt16, UInt64>(mapped, array, res) || || executeType<UInt32, UInt64>(mapped, array, res) || executeType<UInt64, UInt64>(mapped, array, res)
executeType< UInt32, UInt64>(mapped, array, res) || || executeType<UInt128, UInt128>(mapped, array, res) || executeType<UInt256, UInt256>(mapped, array, res)
executeType< UInt64, UInt64>(mapped, array, res) || || executeType<Int8, Int64>(mapped, array, res) || executeType<Int16, Int64>(mapped, array, res)
executeType< Int8 , Int64>(mapped, array, res) || || executeType<Int32, Int64>(mapped, array, res) || executeType<Int64, Int64>(mapped, array, res)
executeType< Int16, Int64>(mapped, array, res) || || executeType<Int128, Int128>(mapped, array, res) || executeType<Int256, Int256>(mapped, array, res)
executeType< Int32, Int64>(mapped, array, res) || || executeType<Float32, Float64>(mapped, array, res) || executeType<Float64, Float64>(mapped, array, res)
executeType< Int64, Int64>(mapped, array, res) || || executeType<Decimal32, Decimal128>(mapped, array, res) || executeType<Decimal64, Decimal128>(mapped, array, res)
executeType<Float32,Float64>(mapped, array, res) || || executeType<Decimal128, Decimal128>(mapped, array, res) || executeType<Decimal256, Decimal256>(mapped, array, res))
executeType<Float64,Float64>(mapped, array, res) ||
executeType<Decimal32, Decimal128>(mapped, array, res) ||
executeType<Decimal64, Decimal128>(mapped, array, res) ||
executeType<Decimal128, Decimal128>(mapped, array, res) ||
executeType<Decimal256, Decimal256>(mapped, array, res))
return res; return res;
else else
throw Exception(ErrorCodes::ILLEGAL_COLUMN, "Unexpected column for arrayCumSumNonNegativeImpl: {}", mapped->getName()); throw Exception(ErrorCodes::ILLEGAL_COLUMN, "Unexpected column for arrayCumSumNonNegativeImpl: {}", mapped->getName());
} }
}; };
struct NameArrayCumSumNonNegative { static constexpr auto name = "arrayCumSumNonNegative"; }; struct NameArrayCumSumNonNegative
{
static constexpr auto name = "arrayCumSumNonNegative";
};
using FunctionArrayCumSumNonNegative = FunctionArrayMapped<ArrayCumSumNonNegativeImpl, NameArrayCumSumNonNegative>; using FunctionArrayCumSumNonNegative = FunctionArrayMapped<ArrayCumSumNonNegativeImpl, NameArrayCumSumNonNegative>;
REGISTER_FUNCTION(ArrayCumSumNonNegative) REGISTER_FUNCTION(ArrayCumSumNonNegative)

9
tests/queries/0_stateless/02716_int256_arrayfunc.reference
View File

@ -8,7 +8,6 @@ Array(UInt128)
Array(Int128) Array(Int128)
Array(UInt256) Array(UInt256)
Array(Int256) Array(Int256)
[0,2] [0,2]
[0,2] [0,2]
[0,2] [0,2]
@ -17,3 +16,11 @@ Array(Int128)
Array(Int128) Array(Int128)
Array(Int256) Array(Int256)
Array(Int256) Array(Int256)
[1,0]
[1,0]
[1,3]
[1,3]
Array(Int128)
Array(Int256)
Array(UInt128)
Array(UInt256)

9
tests/queries/0_stateless/02716_int256_arrayfunc.sql
View File

@ -17,3 +17,12 @@ SELECT toTypeName(arrayDifference([CAST('1', 'UInt128'), 3]));
SELECT toTypeName(arrayDifference([CAST('1', 'Int128'), 3])); SELECT toTypeName(arrayDifference([CAST('1', 'Int128'), 3]));
SELECT toTypeName(arrayDifference([CAST('1', 'UInt256'), 3])); SELECT toTypeName(arrayDifference([CAST('1', 'UInt256'), 3]));
SELECT toTypeName(arrayDifference([CAST('1', 'Int256'), 3])); SELECT toTypeName(arrayDifference([CAST('1', 'Int256'), 3]));
SELECT arrayCumSumNonNegative([CAST('1', 'Int128'), -2]);
SELECT arrayCumSumNonNegative([CAST('1', 'Int256'), -2]);
SELECT arrayCumSumNonNegative([CAST('1', 'UInt128'), 2]);
SELECT arrayCumSumNonNegative([CAST('1', 'UInt256'), 2]);
SELECT toTypeName(arrayCumSumNonNegative([CAST('1', 'Int128'), -2]));
SELECT toTypeName(arrayCumSumNonNegative([CAST('1', 'Int256'), -2]));
SELECT toTypeName(arrayCumSumNonNegative([CAST('1', 'UInt128'), 2]));
SELECT toTypeName(arrayCumSumNonNegative([CAST('1', 'UInt256'), 2]));