95 lines
4.4 KiB
Python
95 lines
4.4 KiB
Python
from wsgiref.headers import tspecials
|
|
from util.log import *
|
|
from util.cases import *
|
|
from util.sql import *
|
|
import numpy as np
|
|
|
|
|
|
class TDTestCase:
|
|
def init(self, conn, logSql, replicaVar=1):
|
|
self.replicaVar = int(replicaVar)
|
|
tdLog.debug("start to execute %s" % __file__)
|
|
tdSql.init(conn.cursor())
|
|
|
|
self.rowNum = 10
|
|
self.batchNum = 5
|
|
self.ts = 1537146000000
|
|
|
|
def groupby_value(self):
|
|
tdSql.query('select 1 from stb group by now')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 1)
|
|
tdSql.query('select 1 from stb group by "1"')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 1)
|
|
tdSql.query('select count(*) from stb group by now')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 12)
|
|
tdSql.query('select count(*) from stb group by now+1')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 12)
|
|
tdSql.query('select 1, count(*) from stb group by now, "1"')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 1)
|
|
tdSql.checkData(0, 1, 12)
|
|
tdSql.query('select count(*) as cc from sta1 as a join sta2 as b on a.ts = b.ts group by now')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 0, 3)
|
|
tdSql.query('select a.tbname, count(*) as cc from sta1 as a join sta2 as b on a.ts = b.ts group by a.tbname, "1"')
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 1, 3)
|
|
|
|
def run(self):
|
|
dbname = "db"
|
|
tdSql.prepare()
|
|
|
|
tdSql.execute(f'''create table sta(ts timestamp, col1 int, col2 bigint) tags(tg1 int, tg2 binary(20))''')
|
|
tdSql.execute(f"create table sta1 using sta tags(1, 'a')")
|
|
tdSql.execute(f"create table sta2 using sta tags(2, 'b')")
|
|
tdSql.execute(f"create table sta3 using sta tags(3, 'c')")
|
|
tdSql.execute(f"create table sta4 using sta tags(4, 'a')")
|
|
tdSql.execute(f"insert into sta1 values(1537146000001, 11, 110)")
|
|
tdSql.execute(f"insert into sta1 values(1537146000002, 12, 120)")
|
|
tdSql.execute(f"insert into sta1 values(1537146000003, 13, 130)")
|
|
tdSql.execute(f"insert into sta2 values(1537146000001, 21, 210)")
|
|
tdSql.execute(f"insert into sta2 values(1537146000002, 22, 220)")
|
|
tdSql.execute(f"insert into sta2 values(1537146000003, 23, 230)")
|
|
tdSql.execute(f"insert into sta3 values(1537146000001, 31, 310)")
|
|
tdSql.execute(f"insert into sta3 values(1537146000002, 32, 320)")
|
|
tdSql.execute(f"insert into sta3 values(1537146000003, 33, 330)")
|
|
tdSql.execute(f"insert into sta4 values(1537146000001, 41, 410)")
|
|
tdSql.execute(f"insert into sta4 values(1537146000002, 42, 420)")
|
|
tdSql.execute(f"insert into sta4 values(1537146000003, 43, 430)")
|
|
|
|
tdSql.execute(f'''create table stb(ts timestamp, col1 int, col2 bigint) tags(tg1 int, tg2 binary(20))''')
|
|
tdSql.execute(f"create table stb1 using stb tags(1, 'a')")
|
|
tdSql.execute(f"create table stb2 using stb tags(2, 'b')")
|
|
tdSql.execute(f"create table stb3 using stb tags(3, 'c')")
|
|
tdSql.execute(f"create table stb4 using stb tags(4, 'a')")
|
|
tdSql.execute(f"insert into stb1 values(1537146000001, 911, 9110)")
|
|
tdSql.execute(f"insert into stb1 values(1537146000002, 912, 9120)")
|
|
tdSql.execute(f"insert into stb1 values(1537146000003, 913, 9130)")
|
|
tdSql.execute(f"insert into stb2 values(1537146000001, 921, 9210)")
|
|
tdSql.execute(f"insert into stb2 values(1537146000002, 922, 9220)")
|
|
tdSql.execute(f"insert into stb2 values(1537146000003, 923, 9230)")
|
|
tdSql.execute(f"insert into stb3 values(1537146000001, 931, 9310)")
|
|
tdSql.execute(f"insert into stb3 values(1537146000002, 932, 9320)")
|
|
tdSql.execute(f"insert into stb3 values(1537146000003, 933, 9330)")
|
|
tdSql.execute(f"insert into stb4 values(1537146000001, 941, 9410)")
|
|
tdSql.execute(f"insert into stb4 values(1537146000002, 942, 9420)")
|
|
tdSql.execute(f"insert into stb4 values(1537146000003, 943, 9430)")
|
|
|
|
tdSql.query("select * from (select ts, col1 from sta partition by tbname) limit 2");
|
|
tdSql.checkRows(2)
|
|
tdSql.query('select col1 > 0 and col2 > 0 from stb')
|
|
tdSql.checkRows(12)
|
|
|
|
self.groupby_value()
|
|
|
|
def stop(self):
|
|
tdSql.close()
|
|
tdLog.success("%s successfully executed" % __file__)
|
|
|
|
tdCases.addWindows(__file__, TDTestCase())
|
|
tdCases.addLinux(__file__, TDTestCase())
|