From 7c2f8bd92df46c183ce034bd99d22505d294c807 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B5=B5=E5=B0=8F=E8=92=99?= Date: Wed, 3 Jul 2024 12:07:08 +0800 Subject: [PATCH] update test_parse_s3path --- tests/test_unit.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_unit.py b/tests/test_unit.py index c5254f82..1dbcf70d 100644 --- a/tests/test_unit.py +++ b/tests/test_unit.py @@ -59,8 +59,9 @@ def test_get_top_percent_list(num_list: list, percent: float, target_num_list: l # 输入一个s3路径，返回bucket名字和其余部分(key) @pytest.mark.parametrize("s3_path, target_data", [ ("s3://bucket/path/to/my/file.txt", "bucket"), - ("/path/to/my/file1.txt", "path"), - ("bucket/path/to/my/file2.txt", "bucket"), + ("s3a://bucket1/path/to/my/file2.txt", "bucket1"), + # ("/path/to/my/file1.txt", "path"), + # ("bucket/path/to/my/file2.txt", "bucket"), ]) def test_parse_s3path(s3_path: str, target_data: str): """ @@ -68,8 +69,8 @@ def test_parse_s3path(s3_path: str, target_data: str): 如果为无效路径，则返回对应的bucket名字和其余部分 如果为异常路径 例如：file2.txt，则报异常 """ - out_keys = parse_s3path(s3_path) - assert target_data == out_keys[0] + bucket_name, key = parse_s3path(s3_path) + assert target_data == bucket_name # 2个box是否处于包含或者部分重合关系。