137 lines
6.9 KiB
Python
137 lines
6.9 KiB
Python
import taos
|
|
import sys
|
|
import datetime
|
|
import inspect
|
|
|
|
from util.log import *
|
|
from util.sql import *
|
|
from util.cases import *
|
|
import random
|
|
|
|
|
|
class TDTestCase:
|
|
|
|
def init(self, conn, logSql, replicaVar=1):
|
|
self.replicaVar = int(replicaVar)
|
|
self.database = "db1"
|
|
tdLog.debug(f"start to excute {__file__}")
|
|
tdSql.init(conn.cursor(), True)
|
|
|
|
def prepare_db(self):
|
|
tdSql.execute(f"drop database if exists {self.database}")
|
|
tdSql.execute(f"create database {self.database}")
|
|
tdSql.execute(f"use {self.database}")
|
|
|
|
def test_value(self, table_name, dtype, bits):
|
|
tdSql.execute(f"drop table if exists {table_name}")
|
|
tdSql.execute(f"create table {table_name}(ts timestamp, i1 {dtype}, i2 {dtype} unsigned)")
|
|
|
|
tdSql.execute(f"insert into {table_name} values(1717122943000, -16, +6)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122944000, 80.99, +0042)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122945000, -0042, +80.99)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122946000, 52.34354, 18.6)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122947000, -12., +3.)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122948000, -0.12, +3.0)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122949000, -2.3e1, +2.324e2)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122950000, -2e1, +2e2)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122951000, -2.e1, +2.e2)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122952000, -0x40, +0b10000)")
|
|
tdSql.execute(f"insert into {table_name} values(1717122953000, -0b10000, +0x40)")
|
|
|
|
# str support
|
|
tdSql.execute(f"insert into {table_name} values(1717122954000, '-16', '+6')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122955000, ' -80.99', ' +0042')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122956000, ' -0042', ' +80.99')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122957000, '52.34354', '18.6')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122958000, '-12.', '+5.')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122959000, '-.12', '+.5')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122960000, '-2.e1', '+2.e2')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122961000, '-2e1', '+2e2')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122962000, '-2.3e1', '+2.324e2')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122963000, '-0x40', '+0b10010')")
|
|
tdSql.execute(f"insert into {table_name} values(1717122964000, '-0b10010', '+0x40')")
|
|
|
|
tdSql.query(f"select * from {table_name}")
|
|
tdSql.checkRows(22)
|
|
|
|
baseval = 2**(bits/2)
|
|
negval = -baseval + 1.645
|
|
posval = baseval + 4.323
|
|
bigval = 2**(bits-1)
|
|
max_i = bigval - 1
|
|
min_i = -bigval
|
|
max_u = 2*bigval - 1
|
|
min_u = 0
|
|
print("val:", baseval, negval, posval, max_i)
|
|
|
|
tdSql.execute(f"insert into {table_name} values(1717122965000, {negval}, {posval})")
|
|
tdSql.execute(f"insert into {table_name} values(1717122966000, -{baseval}, {baseval})")
|
|
tdSql.execute(f"insert into {table_name} values(1717122967000, {max_i}, {max_u})")
|
|
tdSql.execute(f"insert into {table_name} values(1717122968000, {min_i}, {min_u})")
|
|
|
|
tdSql.query(f"select * from {table_name}")
|
|
tdSql.checkRows(26)
|
|
|
|
# error case
|
|
tdSql.error(f"insert into {table_name} values(1717122969000, 0, {max_u+1})")
|
|
tdSql.error(f"insert into {table_name} values(1717122970000, 0, -1)")
|
|
tdSql.error(f"insert into {table_name} values(1717122971000, 0, -2.0)")
|
|
tdSql.error(f"insert into {table_name} values(1717122972000, 0, '-2.0')")
|
|
tdSql.error(f"insert into {table_name} values(1717122973000, {max_i+1}, 0)")
|
|
tdSql.error(f"insert into {table_name} values(1717122974000, {min_i-1}, 0)")
|
|
tdSql.error(f"insert into {table_name} values(1717122975000, '{min_i-1}', 0)")
|
|
|
|
def test_tags(self, stable_name, dtype, bits):
|
|
tdSql.execute(f"create stable {stable_name}(ts timestamp, i1 {dtype}, i2 {dtype} unsigned) tags(id {dtype})")
|
|
|
|
baseval = 2**(bits/2)
|
|
negval = -baseval + 1.645
|
|
posval = baseval + 4.323
|
|
bigval = 2**(bits-1)
|
|
max_i = bigval - 1
|
|
min_i = -bigval
|
|
max_u = 2*bigval - 1
|
|
min_u = 0
|
|
|
|
tdSql.execute(f"insert into {stable_name}_1 using {stable_name} tags('{negval}') values(1717122976000, {negval}, {posval})")
|
|
tdSql.execute(f"insert into {stable_name}_2 using {stable_name} tags({posval}) values(1717122977000, -{baseval} , {baseval})")
|
|
tdSql.execute(f"insert into {stable_name}_3 using {stable_name} tags('0x40') values(1717122978000, {max_i}, {max_u})")
|
|
tdSql.execute(f"insert into {stable_name}_4 using {stable_name} tags(0b10000) values(1717122979000, {min_i}, {min_u})")
|
|
|
|
tdSql.execute(f"insert into {stable_name}_5 using {stable_name} tags({max_i}) values(1717122980000, '{negval}', '{posval}')")
|
|
tdSql.execute(f"insert into {stable_name}_6 using {stable_name} tags('{min_i}') values(1717122981000, '-{baseval}' , '{baseval}')")
|
|
tdSql.execute(f"insert into {stable_name}_7 using {stable_name} tags(-0x40) values(1717122982000, '{max_i}', '{max_u}')")
|
|
tdSql.execute(f"insert into {stable_name}_8 using {stable_name} tags('-0b10000') values(1717122983000, '{min_i}', '{min_u}')")
|
|
|
|
tdSql.execute(f"insert into {stable_name}_9 using {stable_name} tags(12.) values(1717122984000, {negval}, {posval})")
|
|
tdSql.execute(f"insert into {stable_name}_10 using {stable_name} tags('-8.3') values(1717122985000, -{baseval} , {baseval})")
|
|
tdSql.execute(f"insert into {stable_name}_11 using {stable_name} tags(2.e1) values(1717122986000, {max_i}, {max_u})")
|
|
tdSql.execute(f"insert into {stable_name}_12 using {stable_name} tags('-2.3e1') values(1717122987000, {min_i}, {min_u})")
|
|
|
|
tdSql.query(f"select * from {stable_name}")
|
|
tdSql.checkRows(12)
|
|
|
|
def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring
|
|
tdSql.prepare(replica = self.replicaVar)
|
|
self.prepare_db()
|
|
|
|
self.test_value("t1", "bigint", 64)
|
|
self.test_value("t2", "int", 32)
|
|
self.test_value("t3", "smallint", 16)
|
|
self.test_value("t4", "tinyint", 8)
|
|
tdLog.printNoPrefix("==========end case1 run ...............")
|
|
|
|
self.test_tags("t_big", "bigint", 64)
|
|
self.test_tags("t_int", "int", 32)
|
|
self.test_tags("t_small", "smallint", 16)
|
|
self.test_tags("t_tiny", "tinyint", 8)
|
|
tdLog.printNoPrefix("==========end case2 run ...............")
|
|
|
|
def stop(self):
|
|
tdSql.close()
|
|
tdLog.success(f"{__file__} successfully executed")
|
|
|
|
|
|
tdCases.addLinux(__file__, TDTestCase())
|
|
tdCases.addWindows(__file__, TDTestCase())
|