AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-10 21:09:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test: Update tests to use new fixture instead of deprecated one (#7431)

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

Deprecate get_dataset_id_and_document_id fixture, use add_document
instead

### Type of change

- [x] Update test cases
This commit is contained in:
liu an 2025-04-30 14:49:26 +08:00 committed by GitHub
parent e2b0bceb1b
commit e6c824e606
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sdk/python/test/test_http_api/test_chunk_management_within_dataset/test_delete_chunks.py
View File

@ -115,9 +115,9 @@ class TestChunksDeletion:
assert res["data"]["total"] == 1 assert res["data"]["total"] == 1
@pytest.mark.slow @pytest.mark.slow
def test_concurrent_deletion(self, get_http_api_auth, get_dataset_id_and_document_id): def test_concurrent_deletion(self, get_http_api_auth, add_document):
chunks_num = 100 chunks_num = 100
dataset_id, document_id = get_dataset_id_and_document_id dataset_id, document_id = add_document
chunk_ids = batch_add_chunks(get_http_api_auth, dataset_id, document_id, chunks_num) chunk_ids = batch_add_chunks(get_http_api_auth, dataset_id, document_id, chunks_num)
with ThreadPoolExecutor(max_workers=5) as executor: with ThreadPoolExecutor(max_workers=5) as executor:
@ -135,9 +135,9 @@ class TestChunksDeletion:
assert all(r["code"] == 0 for r in responses) assert all(r["code"] == 0 for r in responses)
@pytest.mark.slow @pytest.mark.slow
def test_delete_1k(self, get_http_api_auth, get_dataset_id_and_document_id): def test_delete_1k(self, get_http_api_auth, add_document):
chunks_num = 1_000 chunks_num = 1_000
dataset_id, document_id = get_dataset_id_and_document_id dataset_id, document_id = add_document
chunk_ids = batch_add_chunks(get_http_api_auth, dataset_id, document_id, chunks_num) chunk_ids = batch_add_chunks(get_http_api_auth, dataset_id, document_id, chunks_num)
# issues/6487 # issues/6487