瀏覽代碼

add base unittest s3

renpengli 1 年之前
父節點
當前提交
907d31ae40
共有 1 個文件被更改,包括 9 次插入1 次删除
  1. 9 1
      tests/test_unit.py

+ 9 - 1
tests/test_unit.py

@@ -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))