[TD-2555]<test>add test case for stddev

2021-02-02 18:19:09 +08:00 · 2021-02-02 18:19:09 +08:00 · 8fb87d7faf
parent 751ac814aa
commit 8fb87d7faf
1 changed files with 100 additions and 0 deletions
--- a/tests/pytest/functions/function_stddev_td2555.py
+++ b/tests/pytest/functions/function_stddev_td2555.py
@ -0,0 +1,100 @@
+###################################################################
+#           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 taos
+from util.log import *
+from util.cases import *
+from util.sql import *
+import numpy as np
+
+
+class TDTestCase:
+    def init(self, conn, logSql):
+        tdLog.debug("start to execute %s" % __file__)
+        tdSql.init(conn.cursor())
+
+        self.rowNum = 100
+        self.ts = 1537146000000
+    
+    def getData(self):
+        for i in range(tdSql.queryRows):
+            for j in range(6):
+                exec('self.clist{}.append(tdSql.queryResult[i][j+1])'.format(j+1))
+        
+    def run(self):
+        tdSql.prepare()
+
+        intData = []        
+        floatData = []
+
+        tdSql.execute('''create table test(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 float, col6 double, 
+                    col7 bool, col8 binary(20), col9 nchar(20)) tags(id int,gbid binary(20),loc nchar(20))''')
+        tdSql.execute("create table test1 using test tags(1,'beijing','北京')")
+        tdSql.execute("create table test2 using test tags(2,'shanghai','上海')")
+        tdSql.execute("create table test3 using test tags(3,'shenzhen','深圳')")
+        for j in range(3):
+            for i in range(self.rowNum):
+                tdSql.execute("insert into test%d values(now-%dh, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d')" 
+                            % (j,i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1))
+                intData.append(i + 1)            
+                floatData.append(i + 0.1)                        
+
+        # stddev verifacation 
+        tdSql.error("select stddev(ts) from test")
+        tdSql.error("select stddev(col7) from test")
+        tdSql.error("select stddev(col8) from test")
+        tdSql.error("select stddev(col9) from test")
+
+        tdSql.query("select stddev(col1) from test")
+        tdSql.checkData(0, 0, np.std(intData))
+
+        tdSql.query("select stddev(col2) from test")
+        tdSql.checkData(0, 0, np.std(intData))
+
+        tdSql.query("select stddev(col3) from test")
+        tdSql.checkData(0, 0, np.std(intData))
+
+        tdSql.query("select stddev(col4) from test")
+        tdSql.checkData(0, 0, np.std(intData))
+
+        tdSql.query("select stddev(col5) from test")
+        tdSql.checkData(0, 0, np.std(floatData))
+
+        tdSql.query("select stddev(col6) from test")
+        tdSql.checkData(0, 0, np.std(floatData))
+
+        con_list = [' id = 1 and ts >=now - 1d and ts <now',
+                    ' gbid = ''beijing'' and ts >=now - 1d and ts <now',
+                    ' id = 2 and ts >=now - 1d and ts <now group by tbname',
+                    ' gbid = ''beijing'' and ts >=now - 1d and ts <now group by id ' ,
+                    ' gbid = ''beijing'' and ts >=now - 1d and ts <now group by id ' ,
+                    ' '
+        ]
+        for condition in con_list:
+            tdSql.query("select * from test %s"%(condition))
+            self.getData()
+            for i in range(6):
+                exec('tdSql.query("select stddev(col{}) from test {}")'.format(i+1,condition))
+                exec('tdSql.checkData(0, 0, np.std(self.clist{}))'.format(i+1))
+
+        
+        
+
+            
+    def stop(self):
+        tdSql.close()
+        tdLog.success("%s successfully executed" % __file__)
+
+tdCases.addWindows(__file__, TDTestCase())
+tdCases.addLinux(__file__, TDTestCase())