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

5
sdk/python/test/conftest.py
View File

@ -40,7 +40,10 @@ def login():
@pytest.fixture(scope="session")
def get_api_key_fixture():
register()
try:
register()
except Exception as e:
print(e)
auth = login()
url = HOST_ADDRESS + "/v1/system/new_token"
auth = {"Authorization": auth}

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)

5
sdk/python/test/test_sdk_api/t_chunk.py
View File

@ -190,4 +190,7 @@ def test_retrieve_chunks(get_api_key_fixture):
docs = ds.upload_documents(documents)
doc = docs[0]
doc.add_chunk(content="This is a chunk addition test")
rag.retrieve(dataset_ids=[ds.id],document_ids=[doc.id])
rag.retrieve(dataset_ids=[ds.id],document_ids=[doc.id])
rag.delete_datasets(ids=[ds.id])
# test different parameters for the retrieval