92 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			4.0 KiB
		
	
	
	
		
			Python
		
	
	
	
###################################################################
 | 
						|
#           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 = 10
 | 
						|
        self.ts = 1537146000000
 | 
						|
        
 | 
						|
    def run(self):
 | 
						|
        tdSql.prepare()
 | 
						|
 | 
						|
        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), col11 tinyint unsigned, col12 smallint unsigned, col13 int unsigned, col14 bigint unsigned) tags(loc nchar(20))''')
 | 
						|
        tdSql.execute("create table test1 using test tags('beijing')")
 | 
						|
        for i in range(self.rowNum):
 | 
						|
            tdSql.execute("insert into test1 values(%d, %d, %d, %d, %d, %f, %f, %d, 'taosdata%d', '涛思数据%d', %d, %d, %d, %d)" 
 | 
						|
                        % (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1))                                   
 | 
						|
 | 
						|
        # leastsquares verifacation 
 | 
						|
        tdSql.error("select leastsquares(ts, 1, 1) from test1")
 | 
						|
        tdSql.error("select leastsquares(col1, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col2, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col3, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col4, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col5, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col6, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col7, 1, 1) from test1")
 | 
						|
        tdSql.error("select leastsquares(col8, 1, 1) from test1")
 | 
						|
        tdSql.error("select leastsquares(col9, 1, 1) from test1")
 | 
						|
        tdSql.error("select leastsquares(col11, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col12, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col13, 1, 1) from test")
 | 
						|
        tdSql.error("select leastsquares(col14, 1, 1) from test")
 | 
						|
                
 | 
						|
        tdSql.query("select leastsquares(col1, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col2, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col3, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col4, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col11, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col12, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col13, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col14, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:0.000000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col5, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:-0.900000}')
 | 
						|
 | 
						|
        tdSql.query("select leastsquares(col6, 1, 1) from test1")
 | 
						|
        tdSql.checkData(0, 0, '{slop:1.000000, intercept:-0.900000}')
 | 
						|
 | 
						|
    def stop(self):
 | 
						|
        tdSql.close()
 | 
						|
        tdLog.success("%s successfully executed" % __file__)
 | 
						|
 | 
						|
tdCases.addWindows(__file__, TDTestCase())
 | 
						|
tdCases.addLinux(__file__, TDTestCase())
 |