This is an automated email from the ASF dual-hosted git repository. jfeinauer pushed a commit to branch feature/restrucutre-python-module in repository https://gitbox.apache.org/repos/asf/iotdb.git
commit 5c9e579d9f90a0ed84dfa2ec3fb75ddcd30bfdb4 Author: Julian Feinauer <[email protected]> AuthorDate: Fri Mar 26 09:44:37 2021 +0100 Added first test. --- client-py/iotdb/Session.py | 4 ++++ client-py/{src => }/iotdb/TestContainer.py | 0 client-py/requirements.txt | 5 ++++- client-py/requirements_dev.txt | 5 ++++- client-py/tests/__init__.py | 0 client-py/tests/test_dataframe.py | 23 +++++++++++++++++++---- 6 files changed, 31 insertions(+), 6 deletions(-) diff --git a/client-py/iotdb/Session.py b/client-py/iotdb/Session.py index e371423..e55a692 100644 --- a/client-py/iotdb/Session.py +++ b/client-py/iotdb/Session.py @@ -225,6 +225,10 @@ class Session(object): def insert_str_record(self, device_id, timestamp, measurements, string_values): """ special case for inserting one row of String (TEXT) value """ + if type(string_values) == str: + string_values = [string_values] + if type(measurements) == str: + measurements = [measurements] data_types = [TSDataType.TEXT.value for _ in string_values] request = self.gen_insert_str_record_req(device_id, timestamp, measurements, data_types, string_values) status = self.__client.insertStringRecord(request) diff --git a/client-py/src/iotdb/TestContainer.py b/client-py/iotdb/TestContainer.py similarity index 100% rename from client-py/src/iotdb/TestContainer.py rename to client-py/iotdb/TestContainer.py diff --git a/client-py/requirements.txt b/client-py/requirements.txt index bbd21b4..39c5e61 100644 --- a/client-py/requirements.txt +++ b/client-py/requirements.txt @@ -1 +1,4 @@ -pandas==1.2.3 \ No newline at end of file +# Pandas Export +pandas==1.2.3 +# Testcontainer +testcontainers==3.3.0 \ No newline at end of file diff --git a/client-py/requirements_dev.txt b/client-py/requirements_dev.txt index 8aa50dd..0ee3c7c 100644 --- a/client-py/requirements_dev.txt +++ b/client-py/requirements_dev.txt @@ -1 +1,4 @@ -pytest==6.2.2 \ No newline at end of file +-r requirements.txt +# Pytest to run tests +pytest==6.2.2 +thrift==0.13.0 \ No newline at end of file diff --git a/client-py/tests/__init__.py b/client-py/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/client-py/tests/test_dataframe.py b/client-py/tests/test_dataframe.py index fa41274..988caaf 100644 --- a/client-py/tests/test_dataframe.py +++ b/client-py/tests/test_dataframe.py @@ -1,5 +1,20 @@ -def test_something(): - assert 2 == 1+1 +from iotdb.Session import Session +from iotdb.TestContainer import IoTDBContainer -def test_something_2(): - assert 2 == 1+1+1 \ No newline at end of file + +def test_simple_query(): + with IoTDBContainer("apache/iotdb:0.11.2") as db: + db: IoTDBContainer + session = Session(db.get_container_host_ip(), db.get_exposed_port(6667)) + session.open(False) + + # Write data + session.insert_str_record("device", 123, "pressure", "15.0") + + # Read + session_data_set = session.execute_query_statement("SELECT * FROM root.*") + df = session_data_set.todf() + + session.close() + + assert df == [] \ No newline at end of file
