arctan.py 25.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
import taos
import sys
import datetime
import inspect
import math
from util.log import *
from util.sql import *
from util.cases import *


class TDTestCase:
C
cpwu 已提交
12 13 14
    # updatecfgDict = {'debugFlag': 143 ,"cDebugFlag":143,"uDebugFlag":143 ,"rpcDebugFlag":143 , "tmrDebugFlag":143 ,
    # "jniDebugFlag":143 ,"simDebugFlag":143,"dDebugFlag":143, "dDebugFlag":143,"vDebugFlag":143,"mDebugFlag":143,"qDebugFlag":143,
    # "wDebugFlag":143,"sDebugFlag":143,"tsdbDebugFlag":143,"tqDebugFlag":143 ,"fsDebugFlag":143 ,"udfDebugFlag":143}
15
    def init(self, conn,  logSql, replicaVar=1):
16
        self.replicaVar = int(replicaVar)
17 18
        tdLog.debug(f"start to excute {__file__}")
        tdSql.init(conn.cursor())
G
Ganlin Zhao 已提交
19

C
cpwu 已提交
20
    def prepare_datas(self, dbname="db"):
21
        tdSql.execute(
C
cpwu 已提交
22
            f'''create table {dbname}.stb1
23 24 25 26
            (ts timestamp, c1 int, c2 bigint, c3 smallint, c4 tinyint, c5 float, c6 double, c7 bool, c8 binary(16),c9 nchar(32), c10 timestamp)
            tags (t1 int)
            '''
        )
G
Ganlin Zhao 已提交
27

28
        tdSql.execute(
C
cpwu 已提交
29 30
            f'''
            create table {dbname}.t1
31 32 33 34
            (ts timestamp, c1 int, c2 bigint, c3 smallint, c4 tinyint, c5 float, c6 double, c7 bool, c8 binary(16),c9 nchar(32), c10 timestamp)
            '''
        )
        for i in range(4):
C
cpwu 已提交
35
            tdSql.execute(f'create table {dbname}.ct{i+1} using {dbname}.stb1 tags ( {i+1} )')
36 37 38

        for i in range(9):
            tdSql.execute(
C
cpwu 已提交
39
                f"insert into {dbname}.ct1 values ( now()-{i*10}s, {1*i}, {11111*i}, {111*i}, {11*i}, {1.11*i}, {11.11*i}, {i%2}, 'binary{i}', 'nchar{i}', now()+{1*i}a )"
40 41
            )
            tdSql.execute(
C
cpwu 已提交
42
                f"insert into {dbname}.ct4 values ( now()-{i*90}d, {1*i}, {11111*i}, {111*i}, {11*i}, {1.11*i}, {11.11*i}, {i%2}, 'binary{i}', 'nchar{i}', now()+{1*i}a )"
43
            )
C
cpwu 已提交
44 45 46 47
        tdSql.execute(f"insert into {dbname}.ct1 values (now()-45s, 0, 0, 0, 0, 0, 0, 0, 'binary0', 'nchar0', now()+8a )")
        tdSql.execute(f"insert into {dbname}.ct1 values (now()+10s, 9, -99999, -999, -99, -9.99, -99.99, 1, 'binary9', 'nchar9', now()+9a )")
        tdSql.execute(f"insert into {dbname}.ct1 values (now()+15s, 9, -99999, -999, -99, -9.99, NULL, 1, 'binary9', 'nchar9', now()+9a )")
        tdSql.execute(f"insert into {dbname}.ct1 values (now()+20s, 9, -99999, -999, NULL, -9.99, -99.99, 1, 'binary9', 'nchar9', now()+9a )")
48

C
cpwu 已提交
49 50 51
        tdSql.execute(f"insert into {dbname}.ct4 values (now()-810d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL ) ")
        tdSql.execute(f"insert into {dbname}.ct4 values (now()-400d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL ) ")
        tdSql.execute(f"insert into {dbname}.ct4 values (now()+90d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL  ) ")
52 53

        tdSql.execute(
C
cpwu 已提交
54
            f'''insert into {dbname}.t1 values
55 56 57 58 59 60 61 62 63 64 65 66 67 68
            ( '2020-04-21 01:01:01.000', NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL )
            ( '2020-10-21 01:01:01.000', 1, 11111, 111, 11, 1.11, 11.11, 1, "binary1", "nchar1", now()+1a )
            ( '2020-12-31 01:01:01.000', 2, 22222, 222, 22, 2.22, 22.22, 0, "binary2", "nchar2", now()+2a )
            ( '2021-01-01 01:01:06.000', 3, 33333, 333, 33, 3.33, 33.33, 0, "binary3", "nchar3", now()+3a )
            ( '2021-05-07 01:01:10.000', 4, 44444, 444, 44, 4.44, 44.44, 1, "binary4", "nchar4", now()+4a )
            ( '2021-07-21 01:01:01.000', NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL )
            ( '2021-09-30 01:01:16.000', 5, 55555, 555, 55, 5.55, 55.55, 0, "binary5", "nchar5", now()+5a )
            ( '2022-02-01 01:01:20.000', 6, 66666, 666, 66, 6.66, 66.66, 1, "binary6", "nchar6", now()+6a )
            ( '2022-10-28 01:01:26.000', 7, 00000, 000, 00, 0.00, 00.00, 1, "binary7", "nchar7", "1970-01-01 08:00:00.000" )
            ( '2022-12-01 01:01:30.000', 8, -88888, -888, -88, -8.88, -88.88, 0, "binary8", "nchar8", "1969-01-01 01:00:00.000" )
            ( '2022-12-31 01:01:36.000', 9, -99999999999999999, -999, -99, -9.99, -999999999999999999999.99, 1, "binary9", "nchar9", "1900-01-01 00:00:00.000" )
            ( '2023-02-21 01:01:01.000', NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL )
            '''
        )
G
Ganlin Zhao 已提交
69

70 71 72 73 74 75
    def check_result_auto_atan(self ,origin_query , pow_query):

        pow_result = tdSql.getResult(pow_query)
        origin_result = tdSql.getResult(origin_query)

        auto_result =[]
G
Ganlin Zhao 已提交
76

77 78 79 80 81 82 83 84 85 86
        for row in origin_result:
            row_check = []
            for elem in row:
                if elem == None:
                    elem = None
                else:
                    elem = math.atan(elem)
                row_check.append(elem)
            auto_result.append(row_check)

87
        tdSql.query(pow_query)
88 89
        for row_index , row in enumerate(pow_result):
            for col_index , elem in enumerate(row):
90
                tdSql.checkData(row_index,col_index,auto_result[row_index][col_index])
G
Ganlin Zhao 已提交
91

92
               
C
cpwu 已提交
93
    def test_errors(self, dbname="db"):
94
        error_sql_lists = [
C
cpwu 已提交
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
            f"select atan from {dbname}.t1",
            # f"select atan(-+--+c1 ) from {dbname}.t1",
            # f"select +-atan(c1) from {dbname}.t1",
            # f"select ++-atan(c1) from {dbname}.t1",
            # f"select ++--atan(c1) from {dbname}.t1",
            # f"select - -atan(c1)*0 from {dbname}.t1",
            # f"select atan(tbname+1) from {dbname}.t1 ",
            f"select atan(123--123)==1 from {dbname}.t1",
            f"select atan(c1) as 'd1' from {dbname}.t1",
            f"select atan(c1 ,c2) from {dbname}.t1",
            f"select atan(c1 ,NULL ) from {dbname}.t1",
            f"select atan(,) from {dbname}.t1;",
            f"select atan(atan(c1) ab from {dbname}.t1)",
            f"select atan(c1 ) as int from {dbname}.t1",
            f"select atan from {dbname}.stb1",
            # f"select atan(-+--+c1) from {dbname}.stb1",
            # f"select +-atan(c1) from {dbname}.stb1",
            # f"select ++-atan(c1) from {dbname}.stb1",
            # f"select ++--atan(c1) from {dbname}.stb1",
            # f"select - -atan(c1)*0 from {dbname}.stb1",
            # f"select atan(tbname+1) from {dbname}.stb1 ",
            f"select atan(123--123)==1 from {dbname}.stb1",
            f"select atan(c1) as 'd1' from {dbname}.stb1",
            f"select atan(c1 ,c2 ) from {dbname}.stb1",
            f"select atan(c1 ,NULL) from {dbname}.stb1",
            f"select atan(,) from {dbname}.stb1;",
            f"select atan(atan(c1) ab from {dbname}.stb1)",
            f"select atan(c1) as int from {dbname}.stb1"
123 124 125
        ]
        for error_sql in error_sql_lists:
            tdSql.error(error_sql)
G
Ganlin Zhao 已提交
126

C
cpwu 已提交
127
    def support_types(self, dbname="db"):
128
        type_error_sql_lists = [
C
cpwu 已提交
129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
            f"select atan(ts) from {dbname}.t1" ,
            f"select atan(c7) from {dbname}.t1",
            f"select atan(c8) from {dbname}.t1",
            f"select atan(c9) from {dbname}.t1",
            f"select atan(ts) from {dbname}.ct1" ,
            f"select atan(c7) from {dbname}.ct1",
            f"select atan(c8) from {dbname}.ct1",
            f"select atan(c9) from {dbname}.ct1",
            f"select atan(ts) from {dbname}.ct3" ,
            f"select atan(c7) from {dbname}.ct3",
            f"select atan(c8) from {dbname}.ct3",
            f"select atan(c9) from {dbname}.ct3",
            f"select atan(ts) from {dbname}.ct4" ,
            f"select atan(c7) from {dbname}.ct4",
            f"select atan(c8) from {dbname}.ct4",
            f"select atan(c9) from {dbname}.ct4",
            f"select atan(ts) from {dbname}.stb1" ,
            f"select atan(c7) from {dbname}.stb1",
            f"select atan(c8) from {dbname}.stb1",
            f"select atan(c9) from {dbname}.stb1" ,

            f"select atan(ts) from {dbname}.stbbb1" ,
            f"select atan(c7) from {dbname}.stbbb1",

            f"select atan(ts) from {dbname}.tbname",
            f"select atan(c9) from {dbname}.tbname"
155 156

        ]
G
Ganlin Zhao 已提交
157

158 159
        for type_sql in type_error_sql_lists:
            tdSql.error(type_sql)
G
Ganlin Zhao 已提交
160 161


162
        type_sql_lists = [
C
cpwu 已提交
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192
            f"select atan(c1) from {dbname}.t1",
            f"select atan(c2) from {dbname}.t1",
            f"select atan(c3) from {dbname}.t1",
            f"select atan(c4) from {dbname}.t1",
            f"select atan(c5) from {dbname}.t1",
            f"select atan(c6) from {dbname}.t1",

            f"select atan(c1) from {dbname}.ct1",
            f"select atan(c2) from {dbname}.ct1",
            f"select atan(c3) from {dbname}.ct1",
            f"select atan(c4) from {dbname}.ct1",
            f"select atan(c5) from {dbname}.ct1",
            f"select atan(c6) from {dbname}.ct1",

            f"select atan(c1) from {dbname}.ct3",
            f"select atan(c2) from {dbname}.ct3",
            f"select atan(c3) from {dbname}.ct3",
            f"select atan(c4) from {dbname}.ct3",
            f"select atan(c5) from {dbname}.ct3",
            f"select atan(c6) from {dbname}.ct3",

            f"select atan(c1) from {dbname}.stb1",
            f"select atan(c2) from {dbname}.stb1",
            f"select atan(c3) from {dbname}.stb1",
            f"select atan(c4) from {dbname}.stb1",
            f"select atan(c5) from {dbname}.stb1",
            f"select atan(c6) from {dbname}.stb1",

            f"select atan(c6) as alisb from {dbname}.stb1",
            f"select atan(c6) alisb from {dbname}.stb1",
193 194 195 196
        ]

        for type_sql in type_sql_lists:
            tdSql.query(type_sql)
G
Ganlin Zhao 已提交
197

C
cpwu 已提交
198
    def basic_atan_function(self, dbname="db"):
199

G
Ganlin Zhao 已提交
200
        # basic query
C
cpwu 已提交
201
        tdSql.query(f"select c1 from {dbname}.ct3")
202
        tdSql.checkRows(0)
C
cpwu 已提交
203
        tdSql.query(f"select c1 from {dbname}.t1")
204
        tdSql.checkRows(12)
C
cpwu 已提交
205
        tdSql.query(f"select c1 from {dbname}.stb1")
206 207 208
        tdSql.checkRows(25)

        # used for empty table  , ct3 is empty
C
cpwu 已提交
209
        tdSql.query(f"select atan(c1) from {dbname}.ct3")
210
        tdSql.checkRows(0)
C
cpwu 已提交
211
        tdSql.query(f"select atan(c2) from {dbname}.ct3")
212
        tdSql.checkRows(0)
C
cpwu 已提交
213
        tdSql.query(f"select atan(c3) from {dbname}.ct3")
214
        tdSql.checkRows(0)
C
cpwu 已提交
215
        tdSql.query(f"select atan(c4) from {dbname}.ct3")
216
        tdSql.checkRows(0)
C
cpwu 已提交
217
        tdSql.query(f"select atan(c5) from {dbname}.ct3")
218
        tdSql.checkRows(0)
C
cpwu 已提交
219
        tdSql.query(f"select atan(c6) from {dbname}.ct3")
220 221 222 223
        tdSql.checkRows(0)


        # # used for regular table
C
cpwu 已提交
224
        tdSql.query(f"select atan(c1) from {dbname}.t1")
225 226 227 228 229
        tdSql.checkData(0, 0, None)
        tdSql.checkData(1 , 0, 0.785398163)
        tdSql.checkData(3 , 0, 1.249045772)
        tdSql.checkData(5 , 0, None)

C
cpwu 已提交
230
        tdSql.query(f"select c1, c2, c3 , c4, c5 from {dbname}.t1")
231 232 233 234
        tdSql.checkData(1, 4, 1.11000)
        tdSql.checkData(3, 3, 33)
        tdSql.checkData(5, 4, None)

C
cpwu 已提交
235
        tdSql.query(f"select ts,c1, c2, c3 , c4, c5 from {dbname}.t1")
236 237 238 239
        tdSql.checkData(1, 5, 1.11000)
        tdSql.checkData(3, 4, 33)
        tdSql.checkData(5, 5, None)

C
cpwu 已提交
240
        self.check_result_auto_atan( f"select abs(c1), abs(c2), abs(c3) , abs(c4), abs(c5) from {dbname}.t1", f"select atan(abs(c1)), atan(abs(c2)) ,atan(abs(c3)), atan(abs(c4)), atan(abs(c5)) from {dbname}.t1")
G
Ganlin Zhao 已提交
241

242
        # used for sub table
C
cpwu 已提交
243
        tdSql.query(f"select c2 ,atan(c2) from {dbname}.ct1")
244 245 246
        tdSql.checkData(0, 1, 1.570785077)
        tdSql.checkData(1 , 1, 1.570783470)
        tdSql.checkData(3 , 1, 1.570778327)
C
cpwu 已提交
247
        tdSql.checkData(4 , 1, 0)
248

C
cpwu 已提交
249
        tdSql.query(f"select c1, c5 ,atan(c5) from {dbname}.ct4")
250 251 252 253 254 255
        tdSql.checkData(0 , 2, None)
        tdSql.checkData(1 , 2, 1.458656162)
        tdSql.checkData(2 , 2, 1.442799803)
        tdSql.checkData(3 , 2, 1.421759533)
        tdSql.checkData(5 , 2, None)

C
cpwu 已提交
256
        self.check_result_auto_atan( f"select c1, c2, c3 , c4, c5 from {dbname}.ct1", f"select atan(c1), atan(c2) ,atan(c3), atan(c4), atan(c5) from {dbname}.ct1")
G
Ganlin Zhao 已提交
257

258
        # nest query for atan functions
C
cpwu 已提交
259
        tdSql.query(f"select c4  , atan(c4) ,atan(atan(c4)) , atan(atan(atan(c4))) from {dbname}.ct1;")
260 261 262 263 264 265 266 267 268 269 270 271 272 273 274
        tdSql.checkData(0 , 0 , 88)
        tdSql.checkData(0 , 1 , 1.559433180)
        tdSql.checkData(0 , 2 , 1.000590740)
        tdSql.checkData(0 , 3 , 0.785693446)

        tdSql.checkData(1 , 0 , 77)
        tdSql.checkData(1 , 1 , 1.557810044)
        tdSql.checkData(1 , 2 , 1.000117426)
        tdSql.checkData(1 , 3 , 0.785456873)

        tdSql.checkData(11 , 0 , -99)
        tdSql.checkData(11 , 1 , -1.560695660)
        tdSql.checkData(11 , 2 , -1.000958403)
        tdSql.checkData(11 , 3 , -0.785877135)

G
Ganlin Zhao 已提交
275 276
        # used for stable table

C
cpwu 已提交
277
        tdSql.query(f"select atan(c1) from {dbname}.stb1")
278
        tdSql.checkRows(25)
G
Ganlin Zhao 已提交
279

280 281

        # used for not exists table
C
cpwu 已提交
282 283 284
        tdSql.error(f"select atan(c1) from {dbname}.stbbb1")
        tdSql.error(f"select atan(c1) from {dbname}.tbname")
        tdSql.error(f"select atan(c1) from {dbname}.ct5")
285

G
Ganlin Zhao 已提交
286
        # mix with common col
C
cpwu 已提交
287 288
        tdSql.query(f"select c1, atan(c1) from {dbname}.ct1")
        tdSql.query(f"select c2, atan(c2) from {dbname}.ct4")
G
Ganlin Zhao 已提交
289

290 291

        # mix with common functions
C
cpwu 已提交
292
        tdSql.query(f"select c1, atan(c1),atan(c1), atan(atan(c1)) from {dbname}.ct4 ")
293 294 295 296
        tdSql.checkData(0 , 0 ,None)
        tdSql.checkData(0 , 1 ,None)
        tdSql.checkData(0 , 2 ,None)
        tdSql.checkData(0 , 3 ,None)
G
Ganlin Zhao 已提交
297

298 299 300 301 302
        tdSql.checkData(3 , 0 , 6)
        tdSql.checkData(3 , 1 ,1.405647649)
        tdSql.checkData(3 , 2 ,1.405647649)
        tdSql.checkData(3 , 3 ,0.952449745)

C
cpwu 已提交
303
        tdSql.query(f"select c1, atan(c1),c5, floor(c5) from {dbname}.stb1 ")
304 305

        # # mix with agg functions , not support
C
cpwu 已提交
306 307 308 309 310 311
        tdSql.error(f"select c1, atan(c1),c5, count(c5) from {dbname}.stb1 ")
        tdSql.error(f"select c1, atan(c1),c5, count(c5) from {dbname}.ct1 ")
        tdSql.error(f"select atan(c1), count(c5) from {dbname}.stb1 ")
        tdSql.error(f"select atan(c1), count(c5) from {dbname}.ct1 ")
        tdSql.error(f"select c1, count(c5) from {dbname}.ct1 ")
        tdSql.error(f"select c1, count(c5) from {dbname}.stb1 ")
312 313 314

        # agg functions mix with agg functions

C
cpwu 已提交
315 316
        tdSql.query(f"select max(c5), count(c5) from {dbname}.stb1")
        tdSql.query(f"select max(c5), count(c5) from {dbname}.ct1")
317

G
Ganlin Zhao 已提交
318 319

        # # bug fix for compute
C
cpwu 已提交
320
        tdSql.query(f"select c1, atan(c1) -0 ,atan(c1-4)-0 from {dbname}.ct4 ")
321 322 323 324 325 326 327
        tdSql.checkData(0, 0, None)
        tdSql.checkData(0, 1, None)
        tdSql.checkData(0, 2, None)
        tdSql.checkData(1, 0, 8)
        tdSql.checkData(1, 1, 1.446441332)
        tdSql.checkData(1, 2, 1.325817664)

C
cpwu 已提交
328
        tdSql.query(f" select c1, atan(c1) -0 ,atan(c1-0.1)-0.1 from {dbname}.ct4")
329 330 331 332 333 334 335
        tdSql.checkData(0, 0, None)
        tdSql.checkData(0, 1, None)
        tdSql.checkData(0, 2, None)
        tdSql.checkData(1, 0, 8)
        tdSql.checkData(1, 1, 1.446441332)
        tdSql.checkData(1, 2, 1.344883701)

C
cpwu 已提交
336
        tdSql.query(f"select c1, atan(c1), c2, atan(c2), c3, atan(c3) from {dbname}.ct1")
337

C
cpwu 已提交
338
    def test_big_number(self, dbname="db"):
339

C
cpwu 已提交
340
        tdSql.query(f"select c1, atan(100000000) from {dbname}.ct1")  # bigint to double data overflow
341 342 343
        tdSql.checkData(4, 1, math.atan(100000000))


C
cpwu 已提交
344
        tdSql.query(f"select c1, atan(10000000000000) from {dbname}.ct1")  # bigint to double data overflow
345 346
        tdSql.checkData(4, 1, math.atan(10000000000000))

C
cpwu 已提交
347 348
        tdSql.query(f"select c1, atan(10000000000000000000000000) from {dbname}.ct1")  # bigint to double data overflow
        tdSql.query(f"select c1, atan(10000000000000000000000000.0) from {dbname}.ct1") # 10000000000000000000000000.0 is a double value
349 350
        tdSql.checkData(1, 1, math.atan(10000000000000000000000000.0))

C
cpwu 已提交
351 352
        tdSql.query(f"select c1, atan(10000000000000000000000000000000000) from {dbname}.ct1")  # bigint to double data overflow
        tdSql.query(f"select c1, atan(10000000000000000000000000000000000.0) from {dbname}.ct1") # 10000000000000000000000000.0 is a double value
353 354
        tdSql.checkData(4, 1, math.atan(10000000000000000000000000000000000.0))

C
cpwu 已提交
355 356
        tdSql.query(f"select c1, atan(10000000000000000000000000000000000000000) from {dbname}.ct1")  # bigint to double data overflow
        tdSql.query(f"select c1, atan(10000000000000000000000000000000000000000.0) from {dbname}.ct1") # 10000000000000000000000000.0 is a double value
357 358 359

        tdSql.checkData(4, 1, math.atan(10000000000000000000000000000000000000000.0))

C
cpwu 已提交
360
        tdSql.query(f"select c1, atan(10000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000) from {dbname}.ct1")  # bigint to double data overflow
361

C
cpwu 已提交
362 363 364
    def abs_func_filter(self, dbname="db"):
        tdSql.execute(f"use {dbname}")
        tdSql.query(f"select c1, abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(atan(c1)-0.5) from {dbname}.ct4 where c1>5 ")
365 366 367 368 369 370 371
        tdSql.checkRows(3)
        tdSql.checkData(0,0,8)
        tdSql.checkData(0,1,8.000000000)
        tdSql.checkData(0,2,8.000000000)
        tdSql.checkData(0,3,7.900000000)
        tdSql.checkData(0,4,1.000000000)

C
cpwu 已提交
372
        tdSql.query(f"select c1, abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(atan(c1)-0.5) from {dbname}.ct4 where c1=5 ")
373 374 375 376 377 378 379
        tdSql.checkRows(1)
        tdSql.checkData(0,0,5)
        tdSql.checkData(0,1,5.000000000)
        tdSql.checkData(0,2,5.000000000)
        tdSql.checkData(0,3,4.900000000)
        tdSql.checkData(0,4,1.000000000)

C
cpwu 已提交
380
        tdSql.query(f"select c1,c2 , abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(atan(c1)-0.5) from {dbname}.ct4 where c1=atan(c1) limit 1 ")
381 382 383 384 385 386 387
        tdSql.checkRows(1)
        tdSql.checkData(0,0,0)
        tdSql.checkData(0,1,0)
        tdSql.checkData(0,2,0.000000000)
        tdSql.checkData(0,3,0.000000000)
        tdSql.checkData(0,4,-0.100000000)
        tdSql.checkData(0,5,0.000000000)
G
Ganlin Zhao 已提交
388

389 390
    def pow_Arithmetic(self):
        pass
G
Ganlin Zhao 已提交
391

C
cpwu 已提交
392
    def check_boundary_values(self, dbname="bound_test"):
393 394 395

        PI=3.1415926

C
cpwu 已提交
396 397
        tdSql.execute(f"drop database if exists {dbname}")
        tdSql.execute(f"create database if not exists {dbname}")
398
        time.sleep(3)
C
cpwu 已提交
399
        tdSql.execute(f"use {dbname}")
400
        tdSql.execute(
C
cpwu 已提交
401
            f"create table {dbname}.stb_bound (ts timestamp, c1 int, c2 bigint, c3 smallint, c4 tinyint, c5 float, c6 double, c7 bool, c8 binary(32),c9 nchar(32), c10 timestamp) tags (t1 int);"
402
        )
C
cpwu 已提交
403
        tdSql.execute(f'create table {dbname}.sub1_bound using {dbname}.stb_bound tags ( 1 )')
404
        tdSql.execute(
W
update  
wenzhouwww@live.cn 已提交
405
                f"insert into {dbname}.sub1_bound values ( now()-10s, 2147483647, 9223372036854775807, 32767, 127, 3.40E+38, 1.7e+308, True, 'binary_tb1', 'nchar_tb1', now() )"
406 407
            )
        tdSql.execute(
W
update  
wenzhouwww@live.cn 已提交
408
                f"insert into {dbname}.sub1_bound values ( now()-5s, -2147483647, -9223372036854775807, -32767, -127, -3.40E+38, -1.7e+308, True, 'binary_tb1', 'nchar_tb1', now() )"
409 410
            )
        tdSql.execute(
C
cpwu 已提交
411
                f"insert into {dbname}.sub1_bound values ( now(), 2147483646, 9223372036854775806, 32766, 126, 3.40E+38, 1.7e+308, True, 'binary_tb1', 'nchar_tb1', now() )"
412 413
            )
        tdSql.execute(
W
update  
wenzhouwww@live.cn 已提交
414
                f"insert into {dbname}.sub1_bound values ( now()+5s, -2147483646, -9223372036854775806, -32766, -126, -3.40E+38, -1.7e+308, True, 'binary_tb1', 'nchar_tb1', now() )"
415 416
            )
        tdSql.error(
W
update  
wenzhouwww@live.cn 已提交
417
                f"insert into {dbname}.sub1_bound values ( now()+10s, 2147483648, 9223372036854775808, 32768, 128, 3.40E+38, 1.7e+308, True, 'binary_tb1', 'nchar_tb1', now() )"
418
            )
C
cpwu 已提交
419
        self.check_result_auto_atan( f"select abs(c1), abs(c2), abs(c3) , abs(c4), abs(c5) from {dbname}.sub1_bound ", f"select atan(abs(c1)), atan(abs(c2)) ,atan(abs(c3)), atan(abs(c4)), atan(abs(c5)) from {dbname}.sub1_bound")
G
Ganlin Zhao 已提交
420

C
cpwu 已提交
421
        self.check_result_auto_atan( f"select c1, c2, c3 , c3, c2 ,c1 from {dbname}.sub1_bound ", f"select atan(c1), atan(c2) ,atan(c3), atan(c3), atan(c2) ,atan(c1) from {dbname}.sub1_bound")
422

C
cpwu 已提交
423
        self.check_result_auto_atan(f"select abs(abs(abs(abs(abs(abs(abs(abs(abs(c1)))))))))  nest_col_func from {dbname}.sub1_bound" , f"select atan(abs(c1)) from {dbname}.sub1_bound" )
G
Ganlin Zhao 已提交
424

425
        # check basic elem for table per row
C
cpwu 已提交
426
        tdSql.query(f"select atan(abs(c1)) ,atan(abs(c2)) , atan(abs(c3)) , atan(abs(c4)), atan(abs(c5)), atan(abs(c6)) from {dbname}.sub1_bound ")
427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443
        tdSql.checkData(0,0,math.atan(2147483647))
        tdSql.checkData(0,1,math.atan(9223372036854775807))
        tdSql.checkData(0,2,math.atan(32767))
        tdSql.checkData(0,3,math.atan(127))
        tdSql.checkData(0,4,math.atan(339999995214436424907732413799364296704.00000))
        tdSql.checkData(1,0,math.atan(2147483647))
        tdSql.checkData(1,1,math.atan(9223372036854775807))
        tdSql.checkData(1,2,math.atan(32767))
        tdSql.checkData(1,3,math.atan(127))
        tdSql.checkData(1,4,math.atan(339999995214436424907732413799364296704.00000))
        tdSql.checkData(3,0,math.atan(2147483646))
        tdSql.checkData(3,1,math.atan(9223372036854775806))
        tdSql.checkData(3,2,math.atan(32766))
        tdSql.checkData(3,3,math.atan(126))
        tdSql.checkData(3,4,math.atan(339999995214436424907732413799364296704.00000))

        # check  + - * / in functions
C
cpwu 已提交
444
        tdSql.query(f"select atan(abs(c1+1)) ,atan(abs(c2)) , atan(abs(c3*1)) , atan(abs(c4/2)), atan(abs(c5))/2, atan(abs(c6)) from {dbname}.sub1_bound ")
445 446 447 448 449
        tdSql.checkData(0,0,math.atan(2147483648.000000000))
        tdSql.checkData(0,1,math.atan(9223372036854775807))
        tdSql.checkData(0,2,math.atan(32767.000000000))
        tdSql.checkData(0,3,math.atan(63.500000000))

C
cpwu 已提交
450 451 452 453 454 455 456 457 458 459 460 461 462 463 464
        tdSql.execute(f"create stable {dbname}.st (ts timestamp,  num1 float, num2 double) tags (t1 int);")
        tdSql.execute(f'create table {dbname}.tb1 using {dbname}.st tags (1)')
        tdSql.execute(f'create table {dbname}.tb2 using {dbname}.st tags (2)')
        tdSql.execute(f'create table {dbname}.tb3 using {dbname}.st tags (3)')
        tdSql.execute(f'insert into {dbname}.tb1 values (now()-40s, {PI/2}, {PI/2})')
        tdSql.execute(f'insert into {dbname}.tb1 values (now()-30s, {PI}, {PI})')
        tdSql.execute(f'insert into {dbname}.tb1 values (now()-20s, {PI*1.5}, {PI*1.5})')
        tdSql.execute(f'insert into {dbname}.tb1 values (now()-10s, {PI*2}, {PI*2})')
        tdSql.execute(f'insert into {dbname}.tb1 values (now(), {PI*2.5}, {PI*2.5})')

        tdSql.execute(f'insert into {dbname}.tb2 values (now()-40s, {PI/2}, {PI/2})')
        tdSql.execute(f'insert into {dbname}.tb2 values (now()-30s, {PI}, {PI})')
        tdSql.execute(f'insert into {dbname}.tb2 values (now()-20s, {PI*1.5}, {PI*1.5})')
        tdSql.execute(f'insert into {dbname}.tb2 values (now()-10s, {PI*2}, {PI*2})')
        tdSql.execute(f'insert into {dbname}.tb2 values (now(), {PI*2.5}, {PI*2.5})')
465 466

        for i in range(100):
C
cpwu 已提交
467
            tdSql.execute(f'insert into {dbname}.tb3 values (now()+{i}s, {PI*(5+i)/2}, {PI*(5+i)/2})')
468

C
cpwu 已提交
469
        self.check_result_auto_atan(f"select num1,num2 from {dbname}.tb3;" , f"select atan(num1),atan(num2) from {dbname}.tb3")
470

C
cpwu 已提交
471 472 473 474 475 476
    def support_super_table_test(self, dbname="db"):
        tdSql.execute(f" use {dbname} ")
        self.check_result_auto_atan( f" select c5 from {dbname}.stb1 order by ts " , f"select atan(c5) from {dbname}.stb1 order by ts" )
        self.check_result_auto_atan( f" select c5 from {dbname}.stb1 order by tbname " , f"select atan(c5) from {dbname}.stb1 order by tbname" )
        self.check_result_auto_atan( f" select c5 from {dbname}.stb1 where c1 > 0 order by tbname  " , f"select atan(c5) from {dbname}.stb1 where c1 > 0 order by tbname" )
        self.check_result_auto_atan( f" select c5 from {dbname}.stb1 where c1 > 0 order by tbname  " , f"select atan(c5) from {dbname}.stb1 where c1 > 0 order by tbname" )
477

C
cpwu 已提交
478 479 480 481
        self.check_result_auto_atan( f" select t1,c5 from {dbname}.stb1 order by ts " , f"select atan(t1), atan(c5) from {dbname}.stb1 order by ts" )
        self.check_result_auto_atan( f" select t1,c5 from {dbname}.stb1 order by tbname " , f"select atan(t1) ,atan(c5) from {dbname}.stb1 order by tbname" )
        self.check_result_auto_atan( f" select t1,c5 from {dbname}.stb1 where c1 > 0 order by tbname  " , f"select atan(t1) ,atan(c5) from {dbname}.stb1 where c1 > 0 order by tbname" )
        self.check_result_auto_atan( f" select t1,c5 from {dbname}.stb1 where c1 > 0 order by tbname  " , f"select atan(t1) , atan(c5) from {dbname}.stb1 where c1 > 0 order by tbname" )
482
        pass
G
Ganlin Zhao 已提交
483

C
cpwu 已提交
484

485 486 487 488
    def run(self):  # sourcery skip: extract-duplicate-method, remove-redundant-fstring
        tdSql.prepare()

        tdLog.printNoPrefix("==========step1:create table ==============")
G
Ganlin Zhao 已提交
489

490 491
        self.prepare_datas()

G
Ganlin Zhao 已提交
492
        tdLog.printNoPrefix("==========step2:test errors ==============")
493 494

        self.test_errors()
G
Ganlin Zhao 已提交
495 496

        tdLog.printNoPrefix("==========step3:support types ============")
497 498 499

        self.support_types()

G
Ganlin Zhao 已提交
500
        tdLog.printNoPrefix("==========step4: atan basic query ============")
501 502 503

        self.basic_atan_function()

G
Ganlin Zhao 已提交
504
        tdLog.printNoPrefix("==========step5: big number atan query ============")
505 506 507 508

        self.test_big_number()


G
Ganlin Zhao 已提交
509
        tdLog.printNoPrefix("==========step6: atan boundary query ============")
510 511 512

        self.check_boundary_values()

G
Ganlin Zhao 已提交
513
        tdLog.printNoPrefix("==========step7: atan filter query ============")
514 515 516

        self.abs_func_filter()

C
cpwu 已提交
517
        tdLog.printNoPrefix("==========step8: check atan result of  stable query ============")
518 519 520

        self.support_super_table_test()

G
Ganlin Zhao 已提交
521

522 523 524 525 526 527
    def stop(self):
        tdSql.close()
        tdLog.success(f"{__file__} successfully executed")

tdCases.addLinux(__file__, TDTestCase())
tdCases.addWindows(__file__, TDTestCase())