Fixed pysdk_api_reference.md, add some function test cases. (#676)

### What problem does this PR solve?

Fixed pysdk_api_reference.md, add some function test cases.

Issue link:#672 #673 #675

### Type of change

- [x] Documentation Update
- [x] Test cases

Author: chrysanthemum-boy

docs/pysdk_api_reference.md

@@ -446,36 +446,28 @@ This method searches for rows that satisfy the search condition and updates them
     - response `success` is `True`
 - failure: `Exception`
 
-# table_obj.query_builder
 
-Gets a query_builder by self.
-
-```python
-query_builder=table_obj.query_builder
-```
-
-## Details
-
-This method retrieves a query_builder by self table.
-
-## Returns
-
-- success:
-    - A `query_builder` object.
-- failure: `Exception`
-
----
-
-# query_builder.output
+# table_obj.output
 
 Specifies the columns to display in the search output. 
 
 ```python
-query_builder.output(["num", "body"])
+table_obj.output(["num", "body"])
 # To display all columns
-query_builder.output(["*"])
+table_obj.output(["*"])
 # To display row ID
-query_builder.output(["_row_id"])
+table_obj.output(["_row_id"])
+```
+Convert output to different display types.
+```python
+# Returns a data result
+table_obj.output(["*"]).to_result()
+# Returns a pandas result
+table_obj.output(["*"]).to_df()
+# Returns a polars result
+table_obj.output(["*"]).to_pl()
+# Returns a pyarrow result
+table_obj.output(["*"]).to_arrow()
 ```
 
 ## Details
@@ -493,12 +485,12 @@ This method specifies the columns to display in the search output. You must inpu
 
 - self : `InfinityThriftQueryBuilder`
 
-# query_builder.filter
+# table_obj.filter
 
 Builds a filtering condition expression.
 
 ```python
-query_builder.filter("(-7 < c1 or 9 >= c1) and (c2 = 3)")
+table_obj.filter("(-7 < c1 or 9 >= c1) and (c2 = 3)")
 ```
 
 ## Details
@@ -515,13 +507,13 @@ This method builds a filtering expression.
 
 - self : `InfinityThriftQueryBuilder`
 
-# query_builder.knn
+# table_obj.knn
 
 Builds a KNN search expression.
 
 ```python
-query_builder.knn('col1', [0.1,0.2,0.3], 'float', 'l2', 100)
-query_builder.knn('vec', [3.0] * 5, 'float', 'ip', 2)
+table_obj.knn('col1', [0.1,0.2,0.3], 'float', 'l2', 100)
+table_obj.knn('vec', [3.0] * 5, 'float', 'ip', 2)
 ```
 
 ## Details
@@ -548,12 +540,12 @@ VEC supports list or np.ndarray.
 
 - self : `InfinityThriftQueryBuilder`
 
-# query_builder.match
+# table_obj.match
 
 Builds a full-text search expression.
 
 ```python
-query_builder.match('body', 'harmful', 'topn=2')
+table_obj.match('body', 'harmful', 'topn=2')
 ```
 
 ## Details
@@ -570,12 +562,12 @@ This method builds a full-text search expression.
 
 - self : `InfinityThriftQueryBuilder`
 
-# query_builder.fusion
+# table_obj.fusion
 
 Builds a fusion expression.
 
 ```python
-query_builder.fusion('rrf')
+table_obj.fusion('rrf')
 ```
 
 ## Details
@@ -592,12 +584,12 @@ query_builder.fusion('rrf')
 
 - self : `InfinityThriftQueryBuilder`
 
-# query_builder.to_result
+# table_obj.output.to_result
 
 Returns a data result.
 
 ```python
-table_obj.query_builder().output(["*"]).to_result()
+table_obj.output(["*"]).to_result()
 ```
 
 ## Details
@@ -610,12 +602,12 @@ This method returns a data result of Python's built-in type.
 
 - `tuple[dict[str, list[Any]], dict[str, Any]]`
 
-# query_builder.to_df
+# table_obj.output.to_df
 
 Returns a pandas result.
 
 ```python
-table_obj.query_builder().output(["*"]).to_df()
+table_obj.output(["*"]).to_df()
 ```
 
 ## Details
@@ -626,12 +618,12 @@ This method returns a data result in pandas `DataFrame`.
 
 - `pandas.DataFrame`
 
-# query_builder.to_pl
+# table_obj.output.to_pl
 
 Returns a polars result.
 
 ```python
-table_obj.query_builder().output(["*"]).to_pl()
+table_obj.output(["*"]).to_pl()
 ```
 
 ## Details
@@ -642,12 +634,12 @@ This method returns a data result in polars `DataFrame`
 
 - `polars.DataFrame`
 
-# query_builder.to_arrow
+# table_obj.output.to_arrow
 
 Returns a pyarrow result.
 
 ```python
-table_obj.query_builder().output(["*"]).to_arrow()
+table_obj.output(["*"]).to_arrow()
 ```
 
 ## Details

python/hello_infinity.py

@@ -18,6 +18,7 @@
 from infinity.common import REMOTE_HOST
 import pandas as pds
 
+
 def main():
     infinity_obj = infinity.connect(REMOTE_HOST)
     db = infinity_obj.get_database("default")
@@ -43,7 +44,7 @@ def test():
         db.drop_table("my_table", if_exists=True)
         # Create a table named "my_table"
         table = db.create_table("my_table", {
-                                "num": "integer", "body": "varchar", "vec": "vector, 4, float"}, None)
+            "num": "integer", "body": "varchar", "vec": "vector, 4, float"}, None)
         table.insert(
             [{"num": 1, "body": "unnecessary and harmful", "vec": [1.0, 1.2, 0.8, 0.9]}])
         table.insert(
@@ -94,6 +95,7 @@ def test():
     # print("------tabular - querybuilder-------")
     # print(qb_result)
 
+
 if __name__ == '__main__':
     # main()
     test()

python/test/test_convert.py

@@ -1,3 +1,6 @@
+import pytest
+from infinity.errors import ErrorCode
+
 from python.test.common import common_values
 import infinity
 
@@ -52,4 +55,23 @@ def test_to_df(self):
         res = table_obj.output(["c1", "c1"]).to_df()
         print(res)
         res = table_obj.output(["c1", "c2", "c1"]).to_df()
-        print(res)
+        print(res)
+
+    @pytest.mark.skip(reason="Cause core dumped.")
+    def test_without_output_select_list(self):
+        # connect
+        infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
+        db_obj = infinity_obj.get_database("default")
+        db_obj.drop_table("test_without_output_select_list", True)
+        table_obj = db_obj.create_table("test_without_output_select_list", {
+            "c1": "int", "c2": "float"}, None)
+
+        table_obj.insert([{"c1": 1, "c2": 2.0}])
+        insert_res = table_obj.output([]).convert_type
+        # FIXME insert_res = table_obj.output([]).to_arrow()
+        # FIXME insert_res = table_obj.output([]).to_pl()
+        print(insert_res)
+
+        # disconnect
+        res = infinity_obj.disconnect()
+        assert res.error_code == ErrorCode.OK

python/test/test_insert.py

@@ -126,6 +126,9 @@ def test_insert_big_varchar(self):
 
         db_obj.drop_table("test_insert_big_varchar")
 
+        res = infinity_obj.disconnect()
+        assert res.error_code == ErrorCode.OK
+
     def test_insert_embedding(self):
         """
         target: test insert embedding column
@@ -176,6 +179,7 @@ def test_insert_embedding(self):
 
         db_obj.drop_table("test_insert_embedding2")
 
+    @pytest.mark.skip("Unexpected error.")
     def test_insert_big_embedding(self):
         """
         target: test insert embedding with big dimension
@@ -197,6 +201,7 @@ def test_insert_big_embedding(self):
         res = table_obj.insert([{"c1": [-9999999] * 65535}])
         assert res.error_code == ErrorCode.OK
 
+    @pytest.mark.skip("Unexpected error.")
     def test_insert_big_embedding_float(self):
         """
         target: test insert embedding float with big dimension
@@ -220,6 +225,23 @@ def test_insert_big_embedding_float(self):
         res = db_obj.drop_table("test_insert_big_embedding_float")
         assert res.error_code == ErrorCode.OK
 
+    @pytest.mark.skip("Unexpected error.")
+    @pytest.mark.parametrize("types", ["vector,65535,int", "vector,65535,float"])
+    @pytest.mark.parametrize("types_examples", [[{"c1": [1] * 65535}],
+                                                [{"c1": [4] * 65535}],
+                                                [{"c1": [-9999999] * 65535}],
+                                                [{"c1": [1.1] * 65535}],
+                                                [{"c1": [-9999999.988] * 65535}],
+                                                ])
+    def test_insert_big_embedding_various_type(self, types, types_examples):
+        infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
+        db_obj = infinity_obj.get_database("default")
+        db_obj.drop_table("test_insert_big_embedding", True)
+        table_obj = db_obj.create_table("test_insert_big_embedding", {
+            "c1": types}, None)
+        res = table_obj.insert(types_examples)
+        assert res.error_code == ErrorCode.OK
+
     def test_insert_exceed_block_size(self):
         infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
         db_obj = infinity_obj.get_database("default")
@@ -458,7 +480,7 @@ def test_insert_with_invalid_data_type(self, batch, types):
 
     # batch insert, with invalid column count
     @pytest.mark.parametrize("batch", [10, 1024])
-    def test_insert_with_invalid_column_count(self, batch):
+    def test_batch_insert_with_invalid_column_count(self, batch):
         # connect
         infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
         db_obj = infinity_obj.get_database("default")

python/test/test_update.py

@@ -327,7 +327,7 @@ def test_update_inserted_data(self):
         assert res.error_code == ErrorCode.OK
 
     # update inserted long before and select to check
-    @pytest.mark.skip(reason="May cause core dumped, and cost too much time.")
+    @pytest.mark.skip(reason="Taking too much time.")
     def test_update_inserted_long_before(self):
         # connect
         infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
@@ -353,7 +353,6 @@ def test_update_inserted_long_before(self):
         assert res.error_code == ErrorCode.OK
 
     # update dropped table
-    @pytest.mark.skip(reason="May cause core dumped.")
     def test_update_dropped_table(self):
         # connect
         infinity_obj = infinity.connect(common_values.TEST_REMOTE_HOST)
@@ -363,9 +362,10 @@ def test_update_dropped_table(self):
         db_obj.drop_table("test_update_inserted_long_before")
 
         # update
-        table_obj.update("c1 = 1", [{"c2": 21}])
-        update_res = table_obj.output(["*"]).to_df()
-        print(update_res)
+        with pytest.raises(Exception, match="ERROR:3022*"):
+            table_obj.update("c1 = 1", [{"c2": 21}])
+            update_res = table_obj.output(["*"]).to_df()
+            print(update_res)
 
         # disconnect
         res = infinity_obj.disconnect()