csum.py 21.3 KB
Newer Older
W
wenzhouwww@live.cn 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
###################################################################
#           Copyright (c) 2016 by TAOS Technologies, Inc.
#                     All rights reserved.
#
#  This file is proprietary and confidential to TAOS Technologies.
#  No part of this file may be reproduced, stored, transmitted,
#  disclosed or used in any form or by any means other than as
#  expressly provided by the written permission from Jianhui Tao
#
###################################################################

# -*- coding: utf-8 -*-

import sys
import subprocess
import random
import math
import numpy as np
import inspect
import re

from util.log import *
from util.cases import *
from util.sql import *
from util.dnodes import *


class TDTestCase:
    def init(self, conn, logSql):
        tdLog.debug("start to execute %s" % __file__)
31
        tdSql.init(conn.cursor(), logSql)
W
wenzhouwww@live.cn 已提交
32

W
wenzhouwww@live.cn 已提交
33
    def csum_query_form(self, col="c1",  alias="", table_expr="db.t1", condition=""):
W
wenzhouwww@live.cn 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46

        '''
        csum function:
        :param col:         string, column name, required parameters;
        :param alias:       string, result column another name,or add other funtion;
        :param table_expr:  string or expression, data source(eg,table/stable name, result set), required parameters;
        :param condition:   expression;
        :param args:        other funtions,like: ', last(col)',or give result column another name, like 'c2'
        :return:            csum query statement,default: select csum(c1) from t1
        '''

        return f"select csum({col}) {alias} from {table_expr} {condition}"

W
wenzhouwww@live.cn 已提交
47
    def checkcsum(self,col="c1", alias="", table_expr="db.t1", condition="" ):
W
wenzhouwww@live.cn 已提交
48 49 50 51 52 53 54 55 56 57 58 59 60 61
        line = sys._getframe().f_back.f_lineno
        pre_sql = self.csum_query_form(
            col=col, table_expr=table_expr, condition=condition
        ).replace("csum", "count")
        tdSql.query(pre_sql)

        if tdSql.queryRows == 0:
            tdSql.query(self.csum_query_form(
                col=col, alias=alias, table_expr=table_expr, condition=condition
            ))
            print(f"case in {line}: ", end='')
            tdSql.checkRows(0)
            return

W
wenzhouwww@live.cn 已提交
62 63 64 65 66
        # if "order by tbname" in condition:
        #     tdSql.error(self.csum_query_form(
        #         col=col, alias=alias, table_expr=table_expr, condition=condition
        #     ))
        #     return
W
wenzhouwww@live.cn 已提交
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85

        if "group" in condition:

            tb_condition = condition.split("group by")[1].split(" ")[1]
            tdSql.query(f"select distinct {tb_condition} from {table_expr}")
            query_result = tdSql.queryResult
            query_rows = tdSql.queryRows
            clear_condition = re.sub('order by [0-9a-z]*|slimit [0-9]*|soffset [0-9]*', "", condition)

            pre_row = 0
            for i in range(query_rows):
                group_name = query_result[i][0]
                if "where" in clear_condition:
                    pre_condition = re.sub('group by [0-9a-z]*', f"{tb_condition}='{group_name}'", clear_condition)
                else:
                    pre_condition = "where " + re.sub('group by [0-9a-z]*',f"{tb_condition}='{group_name}'", clear_condition)

                tdSql.query(f"select {col} {alias} from {table_expr} {pre_condition}")
                pre_data = np.array(tdSql.queryResult)[np.array(tdSql.queryResult) != None]
86 87
                if (platform.system().lower() == 'windows' and pre_data.dtype == 'int32'):
                    pre_data = np.array(pre_data, dtype = 'int64')
W
wenzhouwww@live.cn 已提交
88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
                print("data is ", pre_data)
                pre_csum = np.cumsum(pre_data)
                tdSql.query(self.csum_query_form(
                    col=col, alias=alias, table_expr=table_expr, condition=condition
                ))
                for j in range(len(pre_csum)):
                    print(f"case in {line}:", end='')
                    tdSql.checkData(pre_row+j, 1, pre_csum[j])
                pre_row += len(pre_csum)
            return
        elif "union" in condition:
            union_sql_0 = self.csum_query_form(
                col=col, alias=alias, table_expr=table_expr, condition=condition
            ).split("union all")[0]

            union_sql_1 = self.csum_query_form(
                col=col, alias=alias, table_expr=table_expr, condition=condition
            ).split("union all")[1]

            tdSql.query(union_sql_0)
            union_csum_0 = tdSql.queryResult
            row_union_0 = tdSql.queryRows

            tdSql.query(union_sql_1)
            union_csum_1 = tdSql.queryResult

            tdSql.query(self.csum_query_form(
                col=col, alias=alias, table_expr=table_expr, condition=condition
            ))
            for i in range(tdSql.queryRows):
                print(f"case in {line}: ", end='')
                if i < row_union_0:
                    tdSql.checkData(i, 0, union_csum_0[i][0])
                else:
                    tdSql.checkData(i, 0, union_csum_1[i-row_union_0][0])
            return

        else:
W
wenzhouwww@live.cn 已提交
126 127

            tdSql.query(f"select {col} from {table_expr} {re.sub('limit [0-9]*|offset [0-9]*','',condition)} " )
W
wenzhouwww@live.cn 已提交
128 129
            offset_val = condition.split("offset")[1].split(" ")[1] if "offset" in condition else 0
            pre_result = np.array(tdSql.queryResult)[np.array(tdSql.queryResult) != None]
130 131
            if (platform.system().lower() == 'windows' and pre_result.dtype == 'int32'):
                pre_result = np.array(pre_result, dtype = 'int64')
W
wenzhouwww@live.cn 已提交
132 133 134 135 136 137 138
            pre_csum = np.cumsum(pre_result)[offset_val:]
            tdSql.query(self.csum_query_form(
                col=col, alias=alias, table_expr=table_expr, condition=condition
            ))

            for i in range(tdSql.queryRows):
                print(f"case in {line}: ", end='')
W
wenzhouwww@live.cn 已提交
139 140 141 142
                if pre_csum[i] >1.7e+308 or pre_csum[i] < -1.7e+308:
                    continue
                else:
                    tdSql.checkData(i, 0, pre_csum[i])
W
wenzhouwww@live.cn 已提交
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163

        pass

    def csum_current_query(self) :

        # table schema :ts timestamp, c1 int, c2 float, c3 timestamp, c4 binary(16), c5 double, c6 bool
        #                 c7 bigint, c8 smallint, c9 tinyint, c10 nchar(16)

        # case1~6: numeric col:int/bigint/tinyint/smallint/float/double
        self.checkcsum()
        case2 =  {"col": "c2"}
        self.checkcsum(**case2)
        case3 =  {"col": "c5"}
        self.checkcsum(**case3)
        case4 =  {"col": "c7"}
        self.checkcsum(**case4)
        case5 =  {"col": "c8"}
        self.checkcsum(**case5)
        case6 =  {"col": "c9"}
        self.checkcsum(**case6)

W
wenzhouwww@live.cn 已提交
164
        # case7~8: nested query
W
wenzhouwww@live.cn 已提交
165 166 167 168
        case7 = {"table_expr": "(select c1 from db.stb1 order by tbname ,ts )"}
        self.checkcsum(**case7)
        case8 = {"table_expr": "(select csum(c1) c1 from db.t1 partition by tbname)"}
        self.checkcsum(**case8)
W
wenzhouwww@live.cn 已提交
169

W
wenzhouwww@live.cn 已提交
170
        # case9~10: mix with tbname/ts/tag/col not support , must partition by alias  ,such as select tbname ,csum(c1) partition by tbname
W
wenzhouwww@live.cn 已提交
171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
        # case9 = {"alias": ", tbname"}
        # self.checkcsum(**case9)
        # case10 = {"alias": ", _c0"}
        # self.checkcsum(**case10)
        # case11 = {"alias": ", st1"}
        # self.checkcsum(**case11)
        # case12 = {"alias": ", c1"}
        # self.checkcsum(**case12)

        # case13~15: with  single condition
        case13 = {"condition": "where c1 <= 10"}
        self.checkcsum(**case13)
        case14 = {"condition": "where c6 in (0, 1)"}
        self.checkcsum(**case14)
        case15 = {"condition": "where c1 between 1 and 10"}
        self.checkcsum(**case15)

        # case16:  with multi-condition
        case16 = {"condition": "where c6=1 or c6 =0"}
        self.checkcsum(**case16)

        # case17: only support normal table join
        case17 = {
            "col": "t1.c1",
            "table_expr": "t1, t2",
            "condition": "where t1.ts=t2.ts"
        }
        self.checkcsum(**case17)
W
wenzhouwww@live.cn 已提交
199
        # case18~19: with group by
W
wenzhouwww@live.cn 已提交
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
        case18 = {
            "table_expr": "db.t1",
            "condition": "where c6 <0 partition by c6 order by c6"
        }
        self.checkcsum(**case18)
        case19 = {
            "table_expr": "db.t1",
            "condition": " "  # partition by tbname
        }
        self.checkcsum(**case19)

        # case20~21: with order by
        case20 = {"condition": "partition by tbname order by tbname  "}
        # self.checkcsum(**case20) # order by without order by tbname ,because check will random failed

        # case22: with union
        case22 = {
            "condition": "union all select csum(c1) from db.t2"
        }
        # self.checkcsum(**case22) union all without check result becasue ,it will random return table_records
W
wenzhouwww@live.cn 已提交
220 221 222 223 224 225

        # case23: with limit/slimit
        case23 = {
            "condition": "limit 1"
        }
        self.checkcsum(**case23)
W
wenzhouwww@live.cn 已提交
226 227 228 229 230
        case24 = {
            "table_expr": "db.t1",
            "condition": "partition by tbname "
        }
        self.checkcsum(**case24)
W
wenzhouwww@live.cn 已提交
231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247

        pass

    def csum_error_query(self) -> None :
        # unusual test
        #
        # table schema :ts timestamp, c1 int, c2 float, c3 timestamp, c4 binary(16), c5 double, c6 bool
        #                 c7 bigint, c8 smallint, c9 tinyint, c10 nchar(16)
        #
        # form test
        tdSql.error(self.csum_query_form(col=""))   # no col
        tdSql.error("csum(c1) from stb1")           # no select
        tdSql.error("select csum from t1")          # no csum condition
        tdSql.error("select csum c1 from t1")       # no brackets
        tdSql.error("select csum(c1)   t1")         # no from
        tdSql.error("select csum( c1 )  from ")     # no table_expr
        # tdSql.error(self.csum_query_form(col="st1"))    # tag col
W
wenzhouwww@live.cn 已提交
248
        # tdSql.error(self.csum_query_form(col=1))        # col is a value
W
wenzhouwww@live.cn 已提交
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281
        tdSql.error(self.csum_query_form(col="'c1'"))   # col is a string
        tdSql.error(self.csum_query_form(col=None))     # col is NULL 1
        tdSql.error(self.csum_query_form(col="NULL"))   # col is NULL 2
        tdSql.error(self.csum_query_form(col='""'))     # col is ""
        tdSql.error(self.csum_query_form(col='c%'))     # col is spercial char 1
        tdSql.error(self.csum_query_form(col='c_'))     # col is spercial char 2
        tdSql.error(self.csum_query_form(col='c.'))     # col is spercial char 3
        tdSql.error(self.csum_query_form(col='c3'))     # timestamp col
        tdSql.error(self.csum_query_form(col='ts'))     # Primary key
        tdSql.error(self.csum_query_form(col='avg(c1)'))    # expr col
        tdSql.error(self.csum_query_form(col='c6'))     # bool col
        tdSql.error(self.csum_query_form(col='c4'))     # binary col
        tdSql.error(self.csum_query_form(col='c10'))    # nachr col
        tdSql.error(self.csum_query_form(col='c10'))    # not table_expr col
        tdSql.error(self.csum_query_form(col='t1'))     # tbname
        tdSql.error(self.csum_query_form(col='stb1'))   # stbname
        tdSql.error(self.csum_query_form(col='db'))     # datbasename
        tdSql.error(self.csum_query_form(col=True))     # col is BOOL 1
        tdSql.error(self.csum_query_form(col='True'))   # col is BOOL 2
        tdSql.error(self.csum_query_form(col='*'))      # col is all col
        tdSql.error("select csum[c1] from t1")          # sql form error 1
        tdSql.error("select csum{c1} from t1")          # sql form error 2
        tdSql.error(self.csum_query_form(col="[c1]"))   # sql form error 3
        # tdSql.error(self.csum_query_form(col="c1, c2")) # sql form error 3
        # tdSql.error(self.csum_query_form(col="c1, 2"))  # sql form error 3
        tdSql.error(self.csum_query_form(alias=", count(c1)"))  # mix with aggregate function 1
        tdSql.error(self.csum_query_form(alias=", avg(c1)"))    # mix with aggregate function 2
        tdSql.error(self.csum_query_form(alias=", min(c1)"))    # mix with select function 1
        tdSql.error(self.csum_query_form(alias=", top(c1, 5)")) # mix with select function 2
        tdSql.error(self.csum_query_form(alias=", spread(c1)")) # mix with calculation function  1
        tdSql.error(self.csum_query_form(alias=", diff(c1)"))   # mix with calculation function  2
        # tdSql.error(self.csum_query_form(alias=" + 2"))         # mix with arithmetic 1
        tdSql.error(self.csum_query_form(alias=" + avg(c1)"))   # mix with arithmetic 2
G
Ganlin Zhao 已提交
282
        # tdSql.error(self.csum_query_form(alias=", c2"))         # mix with other 1
W
wenzhouwww@live.cn 已提交
283
        # tdSql.error(self.csum_query_form(table_expr="stb1"))    # select stb directly
G
Ganlin Zhao 已提交
284 285 286 287 288 289
        #stb_join = {
        #    "col": "stb1.c1",
        #    "table_expr": "stb1, stb2",
        #    "condition": "where stb1.ts=stb2.ts and stb1.st1=stb2.st2 order by stb1.ts"
        #}
        #tdSql.error(self.csum_query_form(**stb_join))           # stb join
W
wenzhouwww@live.cn 已提交
290 291 292 293 294
        interval_sql = {
            "condition": "where ts>0 and ts < now interval(1h) fill(next)"
        }
        tdSql.error(self.csum_query_form(**interval_sql))       # interval
        group_normal_col = {
W
wenzhouwww@live.cn 已提交
295
            "table_expr": "db.t1",
W
wenzhouwww@live.cn 已提交
296 297 298 299
            "condition": "group by c6"
        }
        tdSql.error(self.csum_query_form(**group_normal_col))       # group by normal col
        slimit_soffset_sql = {
W
wenzhouwww@live.cn 已提交
300
            "table_expr": "db.stb1",
W
wenzhouwww@live.cn 已提交
301 302 303 304
            "condition": "group by tbname slimit 1 soffset 1"
        }
        # tdSql.error(self.csum_query_form(**slimit_soffset_sql))
        order_by_tbname_sql = {
W
wenzhouwww@live.cn 已提交
305
            "table_expr": "db.stb1",
W
wenzhouwww@live.cn 已提交
306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349
            "condition": "group by tbname order by tbname"
        }
        tdSql.error(self.csum_query_form(**order_by_tbname_sql))

        pass

    def csum_test_data(self, tbnum:int, data_row:int, basetime:int) -> None :
        for i in range(tbnum):
            for j in range(data_row):
                tdSql.execute(
                    f"insert into t{i} values ("
                    f"{basetime + (j+1)*10}, {random.randint(-200, -1)}, {random.uniform(200, -1)}, {basetime + random.randint(-200, -1)}, "
                    f"'binary_{j}', {random.uniform(-200, -1)}, {random.choice([0,1])}, {random.randint(-200,-1)}, "
                    f"{random.randint(-200, -1)}, {random.randint(-127, -1)}, 'nchar_{j}' )"
                )

                tdSql.execute(
                    f"insert into t{i} values ("
                    f"{basetime - (j+1) * 10}, {random.randint(1, 200)}, {random.uniform(1, 200)}, {basetime - random.randint(1, 200)}, "
                    f"'binary_{j}_1', {random.uniform(1, 200)}, {random.choice([0, 1])}, {random.randint(1,200)}, "
                    f"{random.randint(1,200)}, {random.randint(1,127)}, 'nchar_{j}_1' )"
                )
                tdSql.execute(
                    f"insert into tt{i} values ( {basetime-(j+1) * 10}, {random.randint(1, 200)} )"
                )

        pass

    def csum_test_table(self,tbnum: int) -> None :
        tdSql.execute("drop database if exists db")
        tdSql.execute("create database  if not exists db keep 3650")
        tdSql.execute("use db")

        tdSql.execute(
            "create stable db.stb1 (\
                ts timestamp, c1 int, c2 float, c3 timestamp, c4 binary(16), c5 double, c6 bool, \
                c7 bigint, c8 smallint, c9 tinyint, c10 nchar(16)\
                ) \
            tags(st1 int)"
        )
        tdSql.execute(
            "create stable db.stb2 (ts timestamp, c1 int) tags(st2 int)"
        )
        for i in range(tbnum):
W
wenzhouwww@live.cn 已提交
350 351
            tdSql.execute(f"create table db.t{i} using db.stb1 tags({i})")
            tdSql.execute(f"create table db.tt{i} using db.stb2 tags({i})")
W
wenzhouwww@live.cn 已提交
352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367

        pass

    def csum_test_run(self) :
        tdLog.printNoPrefix("==========TD-10594==========")
        tbnum = 10
        nowtime = int(round(time.time() * 1000))
        per_table_rows = 2
        self.csum_test_table(tbnum)

        tdLog.printNoPrefix("######## no data test:")
        self.csum_current_query()
        self.csum_error_query()

        tdLog.printNoPrefix("######## insert only NULL test:")
        for i in range(tbnum):
W
wenzhouwww@live.cn 已提交
368 369
            tdSql.execute(f"insert into db.t{i}(ts) values ({nowtime - 5})")
            tdSql.execute(f"insert into db.t{i}(ts) values ({nowtime + 5})")
W
wenzhouwww@live.cn 已提交
370 371 372 373 374
        self.csum_current_query()
        self.csum_error_query()

        tdLog.printNoPrefix("######## insert data in the range near the max(bigint/double):")
        self.csum_test_table(tbnum)
W
wenzhouwww@live.cn 已提交
375
        tdSql.execute(f"insert into db.t1(ts, c1,c2,c5,c7) values "
W
wenzhouwww@live.cn 已提交
376
                      f"({nowtime - (per_table_rows + 1) * 10}, {2**31-1}, {3.4*10**38}, {1.7*10**308}, {2**63-1})")
W
wenzhouwww@live.cn 已提交
377
        tdSql.execute(f"insert into db.t1(ts, c1,c2,c5,c7) values "
W
wenzhouwww@live.cn 已提交
378 379 380 381 382 383
                      f"({nowtime - (per_table_rows + 2) * 10}, {2**31-1}, {3.4*10**38}, {1.7*10**308}, {2**63-1})")
        self.csum_current_query()
        self.csum_error_query()

        tdLog.printNoPrefix("######## insert data in the range near the min(bigint/double):")
        self.csum_test_table(tbnum)
W
wenzhouwww@live.cn 已提交
384
        tdSql.execute(f"insert into db.t1(ts, c1,c2,c5,c7) values "
W
wenzhouwww@live.cn 已提交
385
                      f"({nowtime - (per_table_rows + 1) * 10}, {1-2**31}, {-3.4*10**38}, {-1.7*10**308}, {1-2**63})")
W
wenzhouwww@live.cn 已提交
386
        tdSql.execute(f"insert into db.t1(ts, c1,c2,c5,c7) values "
W
wenzhouwww@live.cn 已提交
387 388 389 390 391 392 393 394 395 396 397 398 399
                      f"({nowtime - (per_table_rows + 2) * 10}, {1-2**31}, {-3.4*10**38}, {-1.7*10**308}, {512-2**63})")
        self.csum_current_query()
        self.csum_error_query()

        tdLog.printNoPrefix("######## insert data without NULL data test:")
        self.csum_test_table(tbnum)
        self.csum_test_data(tbnum, per_table_rows, nowtime)
        self.csum_current_query()
        self.csum_error_query()


        tdLog.printNoPrefix("######## insert data mix with NULL test:")
        for i in range(tbnum):
W
wenzhouwww@live.cn 已提交
400 401 402
            tdSql.execute(f"insert into db.t{i}(ts) values ({nowtime})")
            tdSql.execute(f"insert into db.t{i}(ts) values ({nowtime-(per_table_rows+3)*10})")
            tdSql.execute(f"insert into db.t{i}(ts) values ({nowtime+(per_table_rows+3)*10})")
W
wenzhouwww@live.cn 已提交
403 404 405 406 407 408 409 410 411 412 413 414
        self.csum_current_query()
        self.csum_error_query()



        tdLog.printNoPrefix("######## check after WAL test:")
        tdSql.query("show dnodes")
        index = tdSql.getData(0, 0)
        tdDnodes.stop(index)
        tdDnodes.start(index)
        self.csum_current_query()
        self.csum_error_query()
W
wenzhouwww@live.cn 已提交
415
        tdSql.query("select csum(1) from db.t1 ")
W
wenzhouwww@live.cn 已提交
416 417 418 419 420
        tdSql.checkRows(7)
        tdSql.checkData(0,0,1)
        tdSql.checkData(1,0,2)
        tdSql.checkData(2,0,3)
        tdSql.checkData(3,0,4)
W
wenzhouwww@live.cn 已提交
421
        tdSql.query("select csum(abs(c1))+2 from db.t1 ")
W
wenzhouwww@live.cn 已提交
422
        tdSql.checkRows(4)
G
Ganlin Zhao 已提交
423

G
Ganlin Zhao 已提交
424 425 426 427 428 429 430 431 432 433 434 435 436
        # support selectivity
        tdSql.query("select ts, c1, csum(1) from db.t1")
        tdSql.checkRows(7)

        tdSql.query("select csum(1), ts, c1 from db.t1")
        tdSql.checkRows(7)

        tdSql.query("select ts, c1, c2, c3, csum(1), ts, c4, c5, c6 from db.t1")
        tdSql.checkRows(7)

        tdSql.query("select ts, c1, csum(1), c4, c5, csum(1), c6 from db.t1")
        tdSql.checkRows(7)

437
    def csum_support_stable(self):
W
wenzhouwww@live.cn 已提交
438
        tdSql.query(" select csum(1) from db.stb1 ")
439
        tdSql.checkRows(70)
W
wenzhouwww@live.cn 已提交
440
        tdSql.query("select csum(c1) from db.stb1 partition by tbname ")
441
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
442
        tdSql.query("select csum(st1) from db.stb1 partition by tbname")
443
        tdSql.checkRows(70)
W
wenzhouwww@live.cn 已提交
444
        tdSql.query("select csum(st1+c1) from db.stb1 partition by tbname")
445
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
446
        tdSql.query("select csum(st1+c1) from db.stb1 partition by tbname")
447
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
448
        tdSql.query("select csum(st1+c1) from db.stb1 partition by tbname")
449 450
        tdSql.checkRows(40)

G
Ganlin Zhao 已提交
451
        # # bug need fix
W
wenzhouwww@live.cn 已提交
452
        tdSql.query("select csum(st1+c1) from db.stb1 partition by tbname slimit 1 ")
W
wenzhouwww@live.cn 已提交
453
        tdSql.checkRows(4)
W
wenzhouwww@live.cn 已提交
454
        # tdSql.error("select csum(st1+c1) from db.stb1 partition by tbname limit 1 ")
455 456


G
Ganlin Zhao 已提交
457
        # bug need fix
W
wenzhouwww@live.cn 已提交
458
        tdSql.query("select csum(st1+c1) from db.stb1 partition by tbname")
G
Ganlin Zhao 已提交
459
        tdSql.checkRows(40)
460

G
Ganlin Zhao 已提交
461
        # bug need fix
W
wenzhouwww@live.cn 已提交
462
        tdSql.query("select tbname , csum(c1) from db.stb1 partition by tbname")
463
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
464
        tdSql.query("select tbname , csum(st1) from db.stb1 partition by tbname")
465
        tdSql.checkRows(70)
W
wenzhouwww@live.cn 已提交
466
        tdSql.query("select tbname , csum(st1) from db.stb1 partition by tbname slimit 1")
467
        tdSql.checkRows(7)
468

G
Ganlin Zhao 已提交
469
        # partition by tags
W
wenzhouwww@live.cn 已提交
470
        tdSql.query("select st1 , csum(c1) from db.stb1 partition by st1")
471
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
472
        tdSql.query("select csum(c1) from db.stb1 partition by st1")
473
        tdSql.checkRows(40)
W
wenzhouwww@live.cn 已提交
474
        tdSql.query("select st1 , csum(c1) from db.stb1 partition by st1 slimit 1")
475
        tdSql.checkRows(4)
W
wenzhouwww@live.cn 已提交
476
        tdSql.query("select csum(c1) from db.stb1 partition by st1 slimit 1")
477
        tdSql.checkRows(4)
478 479

        # partition by col
W
wenzhouwww@live.cn 已提交
480
        # tdSql.query("select c1 , csum(c1) from db.stb1 partition by c1")
481
        # tdSql.checkRows(41)
W
wenzhouwww@live.cn 已提交
482
        # tdSql.query("select csum(c1) from db.stb1 partition by c1")
483
        # tdSql.checkRows(41)
W
wenzhouwww@live.cn 已提交
484
        # tdSql.query("select c1 , csum(c1) from db.stb1 partition by st1 slimit 1")
485
        # tdSql.checkRows(4)
W
wenzhouwww@live.cn 已提交
486
        # tdSql.query("select csum(c1) from db.stb1 partition by st1 slimit 1")
487 488
        # tdSql.checkRows(4)

W
wenzhouwww@live.cn 已提交
489

G
Ganlin Zhao 已提交
490

W
wenzhouwww@live.cn 已提交
491 492 493 494 495
    def run(self):
        import traceback
        try:
            # run in  develop branch
            self.csum_test_run()
496
            self.csum_support_stable()
W
wenzhouwww@live.cn 已提交
497 498 499 500 501 502 503 504 505 506 507 508
            pass
        except Exception as e:
            traceback.print_exc()
            raise e



    def stop(self):
        tdSql.close()
        tdLog.success("%s successfully executed" % __file__)

tdCases.addWindows(__file__, TDTestCase())
G
Ganlin Zhao 已提交
509
tdCases.addLinux(__file__, TDTestCase())