pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix test cases (#3718)

Browse Source 
### What problem does this PR solve?

Fix test cases

### Type of change

- [x] Other (please describe): Fix error cases

---------

Signed-off-by: jinhai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2024-11-28 17:37:46 +08:00
committed by GitHub
parent 964a6f4ec4
commit cdae8d28fe
8 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
sdk/python/test/test_frontend_api/test_dataset.py
View File

@ -14,8 +14,8 @@ def test_dataset(get_auth):
dataset_list = []
while True:
res = list_dataset(get_auth, page_number)
data = res.get("data")
for item in data.get("kbs"):
data = res.get("data").get("kbs")
for item in data:
dataset_id = item.get("id")
dataset_list.append(dataset_id)
if len(dataset_list) < page_number * 150:
@ -43,8 +43,8 @@ def test_dataset_1k_dataset(get_auth):
dataset_list = []
while True:
res = list_dataset(get_auth, page_number)
data = res.get("data")
for item in data.get("kbs"):
data = res.get("data").get("kbs")
for item in data:
dataset_id = item.get("id")
dataset_list.append(dataset_id)
if len(dataset_list) < page_number * 150:
@ -66,7 +66,7 @@ def test_duplicated_name_dataset(get_auth):
# list dataset
res = list_dataset(get_auth, 1)
data = res.get("data")
data = res.get("data").get("kbs")
dataset_list = []
pattern = r'^test_create_dataset.*'
for item in data:
@ -109,7 +109,7 @@ def test_update_different_params_dataset(get_auth):
dataset_list = []
while True:
res = list_dataset(get_auth, page_number)
data = res.get("data")
data = res.get("data").get("kbs")
for item in data:
dataset_id = item.get("id")
dataset_list.append(dataset_id)