Skip to content

GitLab

Unverified Commit a6870016 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #1704 from icecraft/fix/ut 

fix: ut
parents 3f22cb60 db2db0c1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/unittest/test_integrations/test_rag/test_utils.py
View file @ a6870016
...@@ -24,7 +24,7 @@ def test_convert_middle_json_to_layout_elements(): ...@@ -24,7 +24,7 @@ def test_convert_middle_json_to_layout_elements():
assert len(res[0].layout_dets) > 0 assert len(res[0].layout_dets) > 0
assert res[0].layout_dets[0].anno_id == 0 assert res[0].layout_dets[0].anno_id == 0
assert res[0].layout_dets[0].category_type == CategoryType.text assert res[0].layout_dets[0].category_type == CategoryType.text
assert len(res[0].extra.element_relation) >= 3 assert len(res[0].extra.element_relation) >= 2
# teardown # teardown
shutil.rmtree(temp_output_dir) shutil.rmtree(temp_output_dir)
...@@ -51,7 +51,7 @@ def test_inference(): ...@@ -51,7 +51,7 @@ def test_inference():
assert len(res[0].layout_dets) > 0 assert len(res[0].layout_dets) > 0
assert res[0].layout_dets[0].anno_id == 0 assert res[0].layout_dets[0].anno_id == 0
assert res[0].layout_dets[0].category_type == CategoryType.text assert res[0].layout_dets[0].category_type == CategoryType.text
assert len(res[0].extra.element_relation) >= 3 assert len(res[0].extra.element_relation) >= 2
# teardown # teardown
shutil.rmtree(temp_output_dir) shutil.rmtree(temp_output_dir)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment