quyuan 1 سال پیش
والد
کامیت
6c5b7ae880
1فایلهای تغییر یافته به همراه4 افزوده شده و 7 حذف شده
  1. 4 7
      tests/test_unit.py

+ 4 - 7
tests/test_unit.py

@@ -1,5 +1,5 @@
 import pytest
-
+import os
 from magic_pdf.libs.boxbase import _is_in_or_part_overlap, _is_in_or_part_overlap_with_area_ratio, _is_in, \
     _is_part_overlap, _left_intersect, _right_intersect, _is_vertical_full_overlap, _is_bottom_full_overlap, \
     _is_left_overlap, __is_overlaps_y_exceeds_threshold, calculate_iou, calculate_overlap_area_2_minbox_area_ratio, \
@@ -516,9 +516,6 @@ def test_bbox_distance(box1: tuple, box2: tuple, target_num: float) -> None:
 
 # 根据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))
+    bucket_name = os.getenv('bucket_name')  
+    target_data = os.getenv('target_data')
+    assert target_data == list(get_s3_config(bucket_name))