zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-23 00:46:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: reformat all code for lefthook using ruff and gofmt (#16585)

Browse Source
This commit is contained in:
Wang Qi
2026-07-03 12:53:39 +08:00
committed by GitHub
parent 19fcb4a981
commit 6a4b9be426
588 changed files with 11123 additions and 15412 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/unit_test/api/utils/test_doc_validation.py
View File

@@ -53,7 +53,7 @@ def test_validate_immutable_fields_no_changes():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -66,7 +66,7 @@ def test_validate_immutable_fields_chunk_count_matches():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -79,7 +79,7 @@ def test_validate_immutable_fields_token_count_matches():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -92,7 +92,7 @@ def test_validate_immutable_fields_progress_matches():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -105,7 +105,7 @@ def test_validate_immutable_fields_chunk_count_mismatch():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg == "Can't change `chunk_count`."
assert error_code == RetCode.DATA_ERROR
@@ -118,7 +118,7 @@ def test_validate_immutable_fields_token_count_mismatch():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg == "Can't change `token_count`."
assert error_code == RetCode.DATA_ERROR
@@ -131,7 +131,7 @@ def test_validate_immutable_fields_progress_mismatch():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg == "Can't change `progress`."
assert error_code == RetCode.DATA_ERROR
@@ -145,18 +145,18 @@ def test_validate_immutable_fields_progress_boundary_values():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.0
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
# Test with 1.0
update_doc_req = UpdateDocumentReq(progress=1.0)
doc = Mock()
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 1.0
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -169,7 +169,7 @@ def test_validate_immutable_fields_none_values():
doc.chunk_num = 10
doc.token_num = 100
doc.progress = 0.5
error_msg, error_code = validate_immutable_fields(update_doc_req, doc)
assert error_msg is None
assert error_code is None
@@ -240,6 +240,7 @@ def test_validate_document_name_valid():
assert error_msg is None
assert error_code is None
def test_validate_document_name_attr_error():
"""Test valid document name update."""
req_doc_name = 0
@@ -312,7 +313,7 @@ def test_validate_chunk_method_valid():
doc = Mock()
doc.type = FileType.PDF
doc.name = "document.pdf"
error_msg, error_code = validate_chunk_method(doc)
assert error_msg is None
assert error_code is None
@@ -323,7 +324,7 @@ def test_validate_chunk_method_visual_not_supported():
doc = Mock()
doc.type = FileType.VISUAL
doc.name = "image.jpg"
error_msg, error_code = validate_chunk_method(doc)
assert "Not supported yet!" in error_msg
assert error_code == RetCode.DATA_ERROR
@@ -334,7 +335,7 @@ def test_validate_chunk_method_ppt_not_supported():
doc = Mock()
doc.type = FileType.PDF
doc.name = "presentation.ppt"
error_msg, error_code = validate_chunk_method(doc)
assert "Not supported yet!" in error_msg
assert error_code == RetCode.DATA_ERROR
@@ -345,7 +346,7 @@ def test_validate_chunk_method_pptx_not_supported():
doc = Mock()
doc.type = FileType.PDF
doc.name = "presentation.pptx"
error_msg, error_code = validate_chunk_method(doc)
assert "Not supported yet!" in error_msg
assert error_code == RetCode.DATA_ERROR
@@ -356,7 +357,7 @@ def test_validate_chunk_method_pages_not_supported():
doc = Mock()
doc.type = FileType.PDF
doc.name = "document.pages"
error_msg, error_code = validate_chunk_method(doc)
assert "Not supported yet!" in error_msg
assert error_code == RetCode.DATA_ERROR
@@ -367,7 +368,7 @@ def test_validate_chunk_method_other_extensions_still_valid():
doc = Mock()
doc.type = FileType.PDF
doc.name = "document.docx"
error_msg, error_code = validate_chunk_method(doc)
assert error_msg is None
assert error_code is None