github.com/whtcorpsinc/milevadb-prod@v0.0.0-20211104133533-f57f4be3b597/interlock/aggfuncs/func_sum_test.go (about)

     1  // Copyright 2020 WHTCORPS INC, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package aggfuncs_test
    15  
    16  import (
    17  	. "github.com/whtcorpsinc/check"
    18  	"github.com/whtcorpsinc/BerolinaSQL/ast"
    19  	"github.com/whtcorpsinc/BerolinaSQL/allegrosql"
    20  	"github.com/whtcorpsinc/milevadb/interlock/aggfuncs"
    21  	"github.com/whtcorpsinc/milevadb/types"
    22  )
    23  
    24  func (s *testSuite) TestMergePartialResult4Sum(c *C) {
    25  	tests := []aggTest{
    26  		builPosetDaggTester(ast.AggFuncSum, allegrosql.TypeNewDecimal, 5, types.NewDecFromInt(10), types.NewDecFromInt(9), types.NewDecFromInt(19)),
    27  		builPosetDaggTester(ast.AggFuncSum, allegrosql.TypeDouble, 5, 10.0, 9.0, 19.0),
    28  	}
    29  	for _, test := range tests {
    30  		s.testMergePartialResult(c, test)
    31  	}
    32  }
    33  
    34  func (s *testSuite) TestSum(c *C) {
    35  	tests := []aggTest{
    36  		builPosetDaggTester(ast.AggFuncSum, allegrosql.TypeNewDecimal, 5, nil, types.NewDecFromInt(10)),
    37  		builPosetDaggTester(ast.AggFuncSum, allegrosql.TypeDouble, 5, nil, 10.0),
    38  	}
    39  	for _, test := range tests {
    40  		s.testAggFunc(c, test)
    41  	}
    42  }
    43  
    44  func (s *testSuite) TestMemSum(c *C) {
    45  	tests := []aggMemTest{
    46  		builPosetDaggMemTester(ast.AggFuncSum, allegrosql.TypeDouble, 5,
    47  			aggfuncs.DefPartialResult4SumFloat64Size, defaultUFIDelateMemDeltaGens, false),
    48  		builPosetDaggMemTester(ast.AggFuncSum, allegrosql.TypeNewDecimal, 5,
    49  			aggfuncs.DefPartialResult4SumDecimalSize, defaultUFIDelateMemDeltaGens, false),
    50  		builPosetDaggMemTester(ast.AggFuncSum, allegrosql.TypeDouble, 5,
    51  			aggfuncs.DefPartialResult4SumDistinctFloat64Size, distinctUFIDelateMemDeltaGens, true),
    52  		builPosetDaggMemTester(ast.AggFuncSum, allegrosql.TypeNewDecimal, 5,
    53  			aggfuncs.DefPartialResult4SumDistinctDecimalSize, distinctUFIDelateMemDeltaGens, true),
    54  	}
    55  	for _, test := range tests {
    56  		s.testAggMemFunc(c, test)
    57  	}
    58  }