Commit 907d31ae authored by renpengli's avatar renpengli

add base unittest s3

parent 4cc005f0
......@@ -7,6 +7,7 @@ from magic_pdf.libs.boxbase import _is_in_or_part_overlap, _is_in_or_part_overla
find_top_nearest_text_bbox, find_bottom_nearest_text_bbox, find_left_nearest_text_bbox, \
find_right_nearest_text_bbox, bbox_relative_pos, bbox_distance
from magic_pdf.libs.commons import mymax, join_path, get_top_percent_list
from magic_pdf.libs.config_reader import get_s3_config
from magic_pdf.libs.path_utils import parse_s3path
......@@ -512,4 +513,11 @@ def test_bbox_relative_pos(box1: tuple, box2: tuple, target_box: tuple) -> None:
def test_bbox_distance(box1: tuple, box2: tuple, target_num: float) -> None:
assert target_num - bbox_distance(box1, box2) < 1
# 根据bucket_name获取s3配置ak,sk,endpoint
def test_get_s3_config() -> None:
with open("./s3_config_testdata.json") as f:
contents = f.read()
for content in eval(contents):
bucket_name = content["bucket_name"]
target_data = content["target_data"]
assert target_data == list(get_s3_config(bucket_name))
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