Skip to content

GitLab

Commit db2db0c1 authored by icecraft's avatar icecraft
Browse files

fix: ut

parent d46b87be
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/unittest/test_integrations/test_rag/test_utils.py
View file @ db2db0c1
...@@ -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