import taos import sys import datetime import inspect import math from util.log import * from util.sql import * from util.cases import * class TDTestCase: 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} def init(self, conn, powSql): tdLog.debug(f"start to excute {__file__}") tdSql.init(conn.cursor()) self.PI =3.1415926 def prepare_datas(self): tdSql.execute( '''create table stb1 (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) ''' ) tdSql.execute( ''' create table t1 (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): tdSql.execute(f'create table ct{i+1} using stb1 tags ( {i+1} )') for i in range(9): tdSql.execute( f"insert into 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 )" ) tdSql.execute( f"insert into 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 )" ) tdSql.execute("insert into ct1 values (now()-45s, 0, 0, 0, 0, 0, 0, 0, 'binary0', 'nchar0', now()+8a )") tdSql.execute("insert into ct1 values (now()+10s, 9, -99999, -999, -99, -9.99, -99.99, 1, 'binary9', 'nchar9', now()+9a )") tdSql.execute("insert into ct1 values (now()+15s, 9, -99999, -999, -99, -9.99, NULL, 1, 'binary9', 'nchar9', now()+9a )") tdSql.execute("insert into ct1 values (now()+20s, 9, -99999, -999, NULL, -9.99, -99.99, 1, 'binary9', 'nchar9', now()+9a )") tdSql.execute("insert into ct4 values (now()-810d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL ) ") tdSql.execute("insert into ct4 values (now()-400d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL ) ") tdSql.execute("insert into ct4 values (now()+90d, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL ) ") tdSql.execute( f'''insert into t1 values ( '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 ) ''' ) def check_result_auto_acos(self ,origin_query , pow_query): pow_result = tdSql.getResult(pow_query) origin_result = tdSql.getResult(origin_query) auto_result =[] for row in origin_result: row_check = [] for elem in row: if elem == None: elem = None elif elem >=-1 and elem <=1: elem = math.acos(elem) else: elem = None row_check.append(elem) auto_result.append(row_check) check_status = True for row_index , row in enumerate(pow_result): for col_index , elem in enumerate(row): if auto_result[row_index][col_index] == None and not (auto_result[row_index][col_index] == None and elem == None): check_status = False elif auto_result[row_index][col_index] != None and (auto_result[row_index][col_index] - elem > 0.00000001): check_status = False else: pass if not check_status: tdLog.notice("acos function value has not as expected , sql is \"%s\" "%pow_query ) sys.exit(1) else: tdLog.info("acos value check pass , it work as expected ,sql is \"%s\" "%pow_query ) def test_errors(self): error_sql_lists = [ "select acos from t1", # "select acos(-+--+c1 ) from t1", # "select +-acos(c1) from t1", # "select ++-acos(c1) from t1", # "select ++--acos(c1) from t1", # "select - -acos(c1)*0 from t1", # "select acos(tbname+1) from t1 ", "select acos(123--123)==1 from t1", "select acos(c1) as 'd1' from t1", "select acos(c1 ,c2) from t1", "select acos(c1 ,NULL ) from t1", "select acos(,) from t1;", "select acos(acos(c1) ab from t1)", "select acos(c1 ) as int from t1", "select acos from stb1", # "select acos(-+--+c1) from stb1", # "select +-acos(c1) from stb1", # "select ++-acos(c1) from stb1", # "select ++--acos(c1) from stb1", # "select - -acos(c1)*0 from stb1", # "select acos(tbname+1) from stb1 ", "select acos(123--123)==1 from stb1", "select acos(c1) as 'd1' from stb1", "select acos(c1 ,c2 ) from stb1", "select acos(c1 ,NULL) from stb1", "select acos(,) from stb1;", "select acos(acos(c1) ab from stb1)", "select acos(c1) as int from stb1" ] for error_sql in error_sql_lists: tdSql.error(error_sql) def support_types(self): type_error_sql_lists = [ "select acos(ts) from t1" , "select acos(c7) from t1", "select acos(c8) from t1", "select acos(c9) from t1", "select acos(ts) from ct1" , "select acos(c7) from ct1", "select acos(c8) from ct1", "select acos(c9) from ct1", "select acos(ts) from ct3" , "select acos(c7) from ct3", "select acos(c8) from ct3", "select acos(c9) from ct3", "select acos(ts) from ct4" , "select acos(c7) from ct4", "select acos(c8) from ct4", "select acos(c9) from ct4", "select acos(ts) from stb1" , "select acos(c7) from stb1", "select acos(c8) from stb1", "select acos(c9) from stb1" , "select acos(ts) from stbbb1" , "select acos(c7) from stbbb1", "select acos(ts) from tbname", "select acos(c9) from tbname" ] for type_sql in type_error_sql_lists: tdSql.error(type_sql) type_sql_lists = [ "select acos(c1) from t1", "select acos(c2) from t1", "select acos(c3) from t1", "select acos(c4) from t1", "select acos(c5) from t1", "select acos(c6) from t1", "select acos(c1) from ct1", "select acos(c2) from ct1", "select acos(c3) from ct1", "select acos(c4) from ct1", "select acos(c5) from ct1", "select acos(c6) from ct1", "select acos(c1) from ct3", "select acos(c2) from ct3", "select acos(c3) from ct3", "select acos(c4) from ct3", "select acos(c5) from ct3", "select acos(c6) from ct3", "select acos(c1) from stb1", "select acos(c2) from stb1", "select acos(c3) from stb1", "select acos(c4) from stb1", "select acos(c5) from stb1", "select acos(c6) from stb1", "select acos(c6) as alisb from stb1", "select acos(c6) alisb from stb1", ] for type_sql in type_sql_lists: tdSql.query(type_sql) def basic_acos_function(self): # basic query tdSql.query("select c1 from ct3") tdSql.checkRows(0) tdSql.query("select c1 from t1") tdSql.checkRows(12) tdSql.query("select c1 from stb1") tdSql.checkRows(25) # used for empty table , ct3 is empty tdSql.query("select acos(c1) from ct3") tdSql.checkRows(0) tdSql.query("select acos(c2) from ct3") tdSql.checkRows(0) tdSql.query("select acos(c3) from ct3") tdSql.checkRows(0) tdSql.query("select acos(c4) from ct3") tdSql.checkRows(0) tdSql.query("select acos(c5) from ct3") tdSql.checkRows(0) tdSql.query("select acos(c6) from ct3") tdSql.checkRows(0) # # used for regular table tdSql.query("select acos(c1) from t1") tdSql.checkData(0, 0, None) tdSql.checkData(1 , 0, 0.000000000) tdSql.checkData(3 , 0, None) tdSql.checkData(5 , 0, None) tdSql.query("select c1, c2, c3 , c4, c5 from t1") tdSql.checkData(1, 4, 1.11000) tdSql.checkData(3, 3, 33) tdSql.checkData(5, 4, None) tdSql.query("select ts,c1, c2, c3 , c4, c5 from t1") tdSql.checkData(1, 5, 1.11000) tdSql.checkData(3, 4, 33) tdSql.checkData(5, 5, None) self.check_result_auto_acos( "select abs(c1), abs(c2), abs(c3) , abs(c4), abs(c5) from t1", "select acos(abs(c1)), acos(abs(c2)) ,acos(abs(c3)), acos(abs(c4)), acos(abs(c5)) from t1") # used for sub table tdSql.query("select c2 ,acos(c2) from ct1") tdSql.checkData(0, 1, None) tdSql.checkData(1 , 1, None) tdSql.checkData(3 , 1, None) tdSql.checkData(4 , 1, 1.570796327) tdSql.query("select c1, c5 ,acos(c5) from ct4") tdSql.checkData(0 , 2, None) tdSql.checkData(1 , 2, None) tdSql.checkData(2 , 2, None) tdSql.checkData(3 , 2, None) tdSql.checkData(5 , 2, None) self.check_result_auto_acos( "select c1, c2, c3 , c4, c5 from ct1", "select acos(c1), acos(c2) ,acos(c3), acos(c4), acos(c5) from ct1") # nest query for acos functions tdSql.query("select c4 , acos(c4) ,acos(acos(c4)) , acos(acos(acos(c4))) from ct1;") tdSql.checkData(0 , 0 , 88) tdSql.checkData(0 , 1 , None) tdSql.checkData(0 , 2 , None) tdSql.checkData(0 , 3 , None) tdSql.checkData(1 , 0 , 77) tdSql.checkData(1 , 1 , None) tdSql.checkData(1 , 2 , None) tdSql.checkData(1 , 3 , None) tdSql.checkData(11 , 0 , -99) tdSql.checkData(11 , 1 , None) tdSql.checkData(11 , 2 , None) tdSql.checkData(11 , 3 , None) # used for stable table tdSql.query("select acos(c1) from stb1") tdSql.checkRows(25) # used for not exists table tdSql.error("select acos(c1) from stbbb1") tdSql.error("select acos(c1) from tbname") tdSql.error("select acos(c1) from ct5") # mix with common col tdSql.query("select c1, acos(c1) from ct1") tdSql.query("select c2, acos(c2) from ct4") # mix with common functions tdSql.query("select c1, acos(c1),acos(c1), acos(acos(c1)) from ct4 ") tdSql.checkData(0 , 0 ,None) tdSql.checkData(0 , 1 ,None) tdSql.checkData(0 , 2 ,None) tdSql.checkData(0 , 3 ,None) tdSql.checkData(3 , 0 , 6) tdSql.checkData(3 , 1 ,None) tdSql.checkData(3 , 2 ,None) tdSql.checkData(3 , 3 ,None) tdSql.query("select c1, acos(c1),c5, floor(c5) from stb1 ") # # mix with agg functions , not support tdSql.error("select c1, acos(c1),c5, count(c5) from stb1 ") tdSql.error("select c1, acos(c1),c5, count(c5) from ct1 ") tdSql.error("select acos(c1), count(c5) from stb1 ") tdSql.error("select acos(c1), count(c5) from ct1 ") tdSql.error("select c1, count(c5) from ct1 ") tdSql.error("select c1, count(c5) from stb1 ") # agg functions mix with agg functions tdSql.query("select max(c5), count(c5) from stb1") tdSql.query("select max(c5), count(c5) from ct1") # # bug fix for compute tdSql.query("select c1, acos(c1) -0 ,acos(c1-4)-0 from ct4 ") tdSql.checkData(0, 0, None) tdSql.checkData(0, 1, None) tdSql.checkData(0, 2, None) tdSql.checkData(1, 0, 8) tdSql.checkData(1, 1, None) tdSql.checkData(1, 2, None) tdSql.query(" select c1, acos(c1) -0 ,acos(c1-0.1)-0.1 from ct4") tdSql.checkData(0, 0, None) tdSql.checkData(0, 1, None) tdSql.checkData(0, 2, None) tdSql.checkData(1, 0, 8) tdSql.checkData(1, 1, None) tdSql.checkData(1, 2, None) tdSql.query("select c1, acos(c1), c2, acos(c2), c3, acos(c3) from ct1") def test_big_number(self): tdSql.query("select c1, acos(100000000) from ct1") # bigint to double data overflow tdSql.checkData(4, 1, None) tdSql.query("select c1, acos(10000000000000) from ct1") # bigint to double data overflow tdSql.checkData(4, 1, None) tdSql.query("select c1, acos(10000000000000000000000000) from ct1") # bigint to double data overflow tdSql.query("select c1, acos(10000000000000000000000000.0) from ct1") # 10000000000000000000000000.0 is a double value tdSql.checkData(1, 1, None) tdSql.query("select c1, acos(10000000000000000000000000000000000) from ct1") # bigint to double data overflow tdSql.query("select c1, acos(10000000000000000000000000000000000.0) from ct1") # 10000000000000000000000000.0 is a double value tdSql.checkData(4, 1, None) tdSql.query("select c1, acos(10000000000000000000000000000000000000000) from ct1") # bigint to double data overflow tdSql.query("select c1, acos(10000000000000000000000000000000000000000.0) from ct1") # 10000000000000000000000000.0 is a double value tdSql.checkData(4, 1, None) tdSql.query("select c1, acos(10000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000) from ct1") # bigint to double data overflow def abs_func_filter(self): tdSql.execute("use db") tdSql.query("select c1, abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(acos(c1)-0.5) from ct4 where c1>5 ") 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,None) tdSql.query("select c1, abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(acos(c1)-0.5) from ct4 where c1=5 ") 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,None) tdSql.query("select c1,c2 , abs(c1) -0 ,ceil(c1-0.1)-0 ,floor(c1+0.1)-0.1 ,ceil(acos(c1)-0.5) from ct4 where c1 0 order by tbname " , "select acos(c5) from stb1 where c1 > 0 order by tbname" ) self.check_result_auto_acos( " select c5 from stb1 where c1 > 0 order by tbname " , "select acos(c5) from stb1 where c1 > 0 order by tbname" ) self.check_result_auto_acos( " select t1,c5 from stb1 order by ts " , "select acos(t1), acos(c5) from stb1 order by ts" ) self.check_result_auto_acos( " select t1,c5 from stb1 order by tbname " , "select acos(t1) ,acos(c5) from stb1 order by tbname" ) self.check_result_auto_acos( " select t1,c5 from stb1 where c1 > 0 order by tbname " , "select acos(t1) ,acos(c5) from stb1 where c1 > 0 order by tbname" ) self.check_result_auto_acos( " select t1,c5 from stb1 where c1 > 0 order by tbname " , "select acos(t1) , acos(c5) from stb1 where c1 > 0 order by tbname" ) pass def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring tdSql.prepare() tdLog.printNoPrefix("==========step1:create table ==============") self.prepare_datas() tdLog.printNoPrefix("==========step2:test errors ==============") self.test_errors() tdLog.printNoPrefix("==========step3:support types ============") self.support_types() tdLog.printNoPrefix("==========step4: acos basic query ============") self.basic_acos_function() tdLog.printNoPrefix("==========step5: big number acos query ============") self.test_big_number() tdLog.printNoPrefix("==========step6: acos boundary query ============") self.check_boundary_values() tdLog.printNoPrefix("==========step7: acos filter query ============") self.abs_func_filter() tdLog.printNoPrefix("==========step7: acos filter query ============") self.abs_func_filter() tdLog.printNoPrefix("==========step8: check acos result of stable query ============") self.support_super_table_test() def stop(self): tdSql.close() tdLog.success(f"{__file__} successfully executed") tdCases.addLinux(__file__, TDTestCase()) tdCases.addWindows(__file__, TDTestCase())