[To rel/0.13] [IOTDB-3494] Fix TypeError in py-session (#6293)
diff --git a/client-py/iotdb/Session.py b/client-py/iotdb/Session.py
index 819a0dd..88af9e8 100644
--- a/client-py/iotdb/Session.py
+++ b/client-py/iotdb/Session.py
@@ -1020,6 +1020,4 @@
"""
if status.code == Session.SUCCESS_CODE:
return 0
-
- logger.error("error status is", status)
return -1
diff --git a/client-py/tests/test_dataframe.py b/client-py/tests/test_dataframe.py
index f95ade6..9a341e1 100644
--- a/client-py/tests/test_dataframe.py
+++ b/client-py/tests/test_dataframe.py
@@ -28,6 +28,7 @@
session = Session(db.get_container_host_ip(), db.get_exposed_port(6667))
session.open(False)
+ session.execute_non_query_statement("set storage group to root.device")
# Write data
session.insert_str_record("root.device", 123, "pressure", "15.0")
@@ -46,7 +47,7 @@
db: IoTDBContainer
session = Session(db.get_container_host_ip(), db.get_exposed_port(6667))
session.open(False)
-
+ session.execute_non_query_statement("set storage group to root.device")
# Write data
session.insert_str_record("root.device", 123, "pressure", "15.0")
@@ -68,8 +69,7 @@
]
assert_array_equal(
df.values,
- [
- [
+ [[
"root.device.pressure",
None,
"root.device",
diff --git a/client-py/tests/test_tablet.py b/client-py/tests/test_tablet.py
index fcb5513..3ad0fdc 100644
--- a/client-py/tests/test_tablet.py
+++ b/client-py/tests/test_tablet.py
@@ -50,6 +50,7 @@
tablet_ = Tablet(
"root.sg_test_01.d_01", measurements_, data_types_, values_, timestamps_
)
+ session.execute_non_query_statement("set storage group to root.sg_test_01")
session.insert_tablet(tablet_)
columns = []
for measurement in measurements_:
@@ -92,6 +93,7 @@
tablet_ = Tablet(
"root.sg_test_01.d_01", measurements_, data_types_, values_, timestamps_
)
+ session.execute_non_query_statement("set storage group to root.sg_test_01")
session.insert_tablet(tablet_)
columns = []
for measurement in measurements_:
diff --git a/client-py/tests/test_todf.py b/client-py/tests/test_todf.py
index feedcb3..8972f29 100644
--- a/client-py/tests/test_todf.py
+++ b/client-py/tests/test_todf.py
@@ -16,10 +16,9 @@
# under the License.
#
-import random
-
import numpy as np
import pandas as pd
+import random
from pandas.testing import assert_frame_equal
from iotdb.IoTDBContainer import IoTDBContainer
@@ -69,6 +68,7 @@
db: IoTDBContainer
session = Session(db.get_container_host_ip(), db.get_exposed_port(6667))
session.open(False)
+ session.execute_non_query_statement("set storage group to root.wt1")
create_ts(session)
@@ -105,7 +105,7 @@
db: IoTDBContainer
session = Session(db.get_container_host_ip(), db.get_exposed_port(6667))
session.open(False)
-
+ session.execute_non_query_statement("set storage group to root.wt1")
create_ts(session)
# insert data
@@ -184,6 +184,7 @@
db: IoTDBContainer
session = Session(db.get_container_host_ip(), db.get_exposed_port(6667))
session.open(False)
+ session.execute_non_query_statement("set storage group to root.wt1")
create_ts(session)