test: remove chunk_keys

Author: taskingaijc

test/testcase/test_async/test_async_retrieval.py

@@ -305,18 +305,6 @@ async def test_a_delete_record(self):
 
 @pytest.mark.test_async
 class TestChunk(Base):
-    chunk_list = [
-        "chunk_id",
-        "record_id",
-        "collection_id",
-        "content",
-        "metadata",
-        "num_tokens",
-        "score",
-        "updated_timestamp",
-        "created_timestamp",
-    ]
-    chunk_keys = set(chunk_list)
 
     @pytest.mark.run(order=41)
     @pytest.mark.asyncio
@@ -332,7 +320,6 @@ async def test_a_query_chunks(self):
         for chunk in res:
             chunk_dict = vars(chunk)
             assume_query_chunk_result(query_text, chunk_dict)
-            pytest.assume(chunk_dict.keys() == self.chunk_keys)
             pytest.assume(chunk_dict["score"] >= 0.04)
 
     @pytest.mark.run(order=42)
@@ -345,7 +332,6 @@ async def test_create_chunk(self):
         }
         res = await a_create_chunk(**create_chunk_data)
         res_dict = vars(res)
-        pytest.assume(res_dict.keys() == self.chunk_keys)
         assume_chunk_result(create_chunk_data, res_dict)
         Base.chunk_id = res_dict["chunk_id"]
 
@@ -386,7 +372,6 @@ async def test_get_chunk(self):
             res_dict = vars(res)
             pytest.assume(res_dict["collection_id"] == self.collection_id)
             pytest.assume(res_dict["chunk_id"] == chunk_id)
-            pytest.assume(res_dict.keys() == self.chunk_keys)
 
     @pytest.mark.run(order=45)
     @pytest.mark.asyncio
@@ -401,7 +386,6 @@ async def test_update_chunk(self):
         }
         res = await a_update_chunk(**update_chunk_data)
         res_dict = vars(res)
-        pytest.assume(res_dict.keys() == self.chunk_keys)
         assume_chunk_result(update_chunk_data, res_dict)
 
     @pytest.mark.run(order=46)

test/testcase/test_sync/test_sync_retrieval.py

@@ -315,18 +315,6 @@ def test_delete_record(self, collection_id):
 
 @pytest.mark.test_sync
 class TestChunk:
-    chunk_list = [
-        "chunk_id",
-        "record_id",
-        "collection_id",
-        "content",
-        "metadata",
-        "num_tokens",
-        "score",
-        "updated_timestamp",
-        "created_timestamp",
-    ]
-    chunk_keys = set(chunk_list)
 
     @pytest.mark.run(order=41)
     def test_query_chunks(self, collection_id):
@@ -341,7 +329,6 @@ def test_query_chunks(self, collection_id):
         for chunk in res:
             chunk_dict = vars(chunk)
             assume_query_chunk_result(query_text, chunk_dict)
-            pytest.assume(chunk_dict.keys() == self.chunk_keys)
             pytest.assume(chunk_dict["score"] >= 0.04)
 
     @pytest.mark.run(order=42)
@@ -353,7 +340,6 @@ def test_create_chunk(self, collection_id):
         }
         res = create_chunk(**create_chunk_data)
         res_dict = vars(res)
-        pytest.assume(res_dict.keys() == self.chunk_keys)
         assume_chunk_result(create_chunk_data, res_dict)
 
     @pytest.mark.run(order=43)
@@ -391,7 +377,6 @@ def test_get_chunk(self, collection_id):
             res_dict = vars(res)
             pytest.assume(res_dict["collection_id"] == collection_id)
             pytest.assume(res_dict["chunk_id"] == chunk_id)
-            pytest.assume(res_dict.keys() == self.chunk_keys)
 
     @pytest.mark.run(order=45)
     def test_update_chunk(self, collection_id, chunk_id):
@@ -405,7 +390,6 @@ def test_update_chunk(self, collection_id, chunk_id):
         }
         res = update_chunk(**update_chunk_data)
         res_dict = vars(res)
-        pytest.assume(res_dict.keys() == self.chunk_keys)
         assume_chunk_result(update_chunk_data, res_dict)
 
     @pytest.mark.run(order=46)