Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
冷面1992
TDengine
提交
d782a83b
TDengine
项目概览
冷面1992
/
TDengine
与 Fork 源项目一致
Fork自
taosdata / TDengine
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
TDengine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
d782a83b
编写于
11月 30, 2020
作者:
H
huili
提交者:
GitHub
11月 30, 2020
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #4379 from taosdata/xiaoping/add_test_case2
[TD-2109] <test> add test case
上级
98a86b3c
cb899a2c
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
118 addition
and
0 deletion
+118
-0
tests/pytest/insert/insertDemo.py
tests/pytest/insert/insertDemo.py
+47
-0
tests/pytest/insert/insertFromCSVOurofOrder.py
tests/pytest/insert/insertFromCSVOurofOrder.py
+71
-0
未找到文件。
tests/pytest/insert/insertDemo.py
0 → 100644
浏览文件 @
d782a83b
import
taos
import
datetime
import
random
import
multiprocessing
def
taos_excute
(
table
,
connect_host
):
conn
=
taos
.
connect
(
host
=
connect_host
,
user
=
"root"
,
password
=
"taosdata"
,
config
=
"/etc/taos"
,
database
=
'test'
)
cursor
=
conn
.
cursor
()
for
i
in
range
(
1000000
):
pk
=
random
.
randint
(
100001
,
300000
)
time_now
=
datetime
.
datetime
.
now
().
strftime
(
"%Y-%m-%d %H:%M:%S.%f"
)[:
-
3
]
col1
=
random
.
randint
(
1
,
10000
)
col2
=
random
.
randint
(
1
,
10000
)
col3
=
random
.
randint
(
1
,
10000
)
col4
=
random
.
randint
(
1
,
10000
)
col5
=
random
.
randint
(
1
,
10000
)
col6
=
random
.
randint
(
1
,
10000
)
sql
=
f
"INSERT INTO
{
table
}
_
{
pk
}
USING
{
table
}
TAGS (
{
pk
}
) VALUES ('
{
time_now
}
',
{
col1
}
,
{
col2
}
,
{
col3
}
,
{
col4
}
,
{
col5
}
,
{
col6
}
)"
cursor
.
execute
(
sql
)
cursor
.
close
()
conn
.
close
()
def
taos_init
(
table
,
connect_host
,
pk
):
conn
=
taos
.
connect
(
host
=
connect_host
,
user
=
"root"
,
password
=
"taosdata"
,
config
=
"/etc/taos"
,
database
=
'test'
)
cursor
=
conn
.
cursor
()
sql
=
f
"CREATE TABLE
{
table
}
_
{
pk
}
USING
{
table
}
TAGS (
{
pk
}
)"
cursor
.
execute
(
sql
)
cursor
.
close
()
conn
.
close
()
print
(
"init time:"
,
datetime
.
datetime
.
now
().
strftime
(
"%Y-%m-%d %H:%M:%S"
))
connect_list
=
[
"node1"
,
"node2"
,
"node3"
,
"node4"
,
"node5"
]
pool
=
multiprocessing
.
Pool
(
processes
=
108
)
for
pk
in
range
(
100001
,
300000
):
pool
.
apply_async
(
func
=
taos_init
,
args
=
(
"test"
,
connect_list
[
pk
%
5
],
pk
,
))
print
(
"start time:"
,
datetime
.
datetime
.
now
().
strftime
(
"%Y-%m-%d %H:%M:%S"
))
for
i
in
range
(
10000
):
pool
.
apply_async
(
func
=
taos_excute
,
args
=
(
"test"
,
connect_list
[
i
%
5
],))
pool
.
close
()
pool
.
join
()
print
(
"end time:"
,
datetime
.
datetime
.
now
().
strftime
(
"%Y-%m-%d %H:%M:%S"
))
\ No newline at end of file
tests/pytest/insert/insertFromCSVOurofOrder.py
0 → 100644
浏览文件 @
d782a83b
###################################################################
# 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
tdLog
from
util.cases
import
tdCases
from
util.sql
import
tdSql
import
time
import
datetime
import
csv
import
random
import
pandas
as
pd
class
TDTestCase
:
def
init
(
self
,
conn
,
logSql
):
tdLog
.
debug
(
"start to execute %s"
%
__file__
)
tdSql
.
init
(
conn
.
cursor
(),
logSql
)
self
.
ts
=
1500074556514
def
writeCSV
(
self
):
with
open
(
'test3.csv'
,
'w'
,
encoding
=
'utf-8'
,
newline
=
''
)
as
csvFile
:
writer
=
csv
.
writer
(
csvFile
,
dialect
=
'excel'
)
for
i
in
range
(
1000000
):
newTimestamp
=
self
.
ts
+
random
.
randint
(
10000000
,
10000000000
)
+
random
.
randint
(
1000
,
10000000
)
+
random
.
randint
(
1
,
1000
)
d
=
datetime
.
datetime
.
fromtimestamp
(
newTimestamp
/
1000
)
dt
=
str
(
d
.
strftime
(
"%Y-%m-%d %H:%M:%S.%f"
))
writer
.
writerow
([
"'%s'"
%
dt
,
random
.
randint
(
1
,
100
),
random
.
uniform
(
1
,
100
),
random
.
randint
(
1
,
100
),
random
.
randint
(
1
,
100
)])
def
removCSVHeader
(
self
):
data
=
pd
.
read_csv
(
"ordered.csv"
)
data
=
data
.
drop
([
0
])
data
.
to_csv
(
"ordered.csv"
,
header
=
False
,
index
=
False
)
def
run
(
self
):
tdSql
.
prepare
()
tdSql
.
execute
(
"create table t1(ts timestamp, c1 int, c2 float, c3 int, c4 int)"
)
startTime
=
time
.
time
()
tdSql
.
execute
(
"insert into t1 file 'outoforder.csv'"
)
duration
=
time
.
time
()
-
startTime
print
(
"Out of Order - Insert time: %d"
%
duration
)
tdSql
.
query
(
"select count(*) from t1"
)
rows
=
tdSql
.
getData
(
0
,
0
)
tdSql
.
execute
(
"create table t2(ts timestamp, c1 int, c2 float, c3 int, c4 int)"
)
startTime
=
time
.
time
()
tdSql
.
execute
(
"insert into t2 file 'ordered.csv'"
)
duration
=
time
.
time
()
-
startTime
print
(
"Ordered - Insert time: %d"
%
duration
)
tdSql
.
query
(
"select count(*) from t2"
)
tdSql
.
checkData
(
0
,
0
,
rows
)
def
stop
(
self
):
tdSql
.
close
()
tdLog
.
success
(
"%s successfully executed"
%
__file__
)
tdCases
.
addWindows
(
__file__
,
TDTestCase
())
tdCases
.
addLinux
(
__file__
,
TDTestCase
())
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录