From c71b1440cf1501218dfa6c7ca3896ad52fd65f6c Mon Sep 17 00:00:00 2001
From: cpwu <cpwu@taosdata.com>
Date: Tue, 26 Jul 2022 15:38:54 +0800
Subject: [PATCH] fix case

---
 tests/pytest/util/common.py               | 2 +-
 tests/system-test/1-insert/mutil_stage.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/pytest/util/common.py b/tests/pytest/util/common.py
index a290a92876..6384195a1b 100644
--- a/tests/pytest/util/common.py
+++ b/tests/pytest/util/common.py
@@ -18,7 +18,7 @@ import time
 import socket
 import json
 import toml
-from .boundary import DataBoundary
+from util.boundary import DataBoundary
 import taos
 from util.log import *
 from util.sql import *
diff --git a/tests/system-test/1-insert/mutil_stage.py b/tests/system-test/1-insert/mutil_stage.py
index 6584c4081b..e4342eca24 100644
--- a/tests/system-test/1-insert/mutil_stage.py
+++ b/tests/system-test/1-insert/mutil_stage.py
@@ -210,7 +210,8 @@ class TDTestCase:
             tdSql.execute(f'create table {dbname}.{ctb_pre}{i+1} using {dbname}.{stb} tags ( {i+1} )')
 
     def __insert_data(self, rows,dbname=DBNAME):
-        data = DataSet().get_order_set(rows)
+        data = DataSet()
+        data.get_order_set(rows)
 
         tdLog.printNoPrefix("==========step: start inser data into tables now.....")
         for i in range(self.rows):