From 47f2bbc7a8b6ef21089a914ce113f1d70719b1bb Mon Sep 17 00:00:00 2001 From: GeneLiuXe <356340460@qq.com> Date: Wed, 6 Dec 2023 11:46:12 +0800 Subject: [PATCH] [FIX] fit with client --- tests/test_learnware_client/test_all_learnware.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/tests/test_learnware_client/test_all_learnware.py b/tests/test_learnware_client/test_all_learnware.py index 276ac00..69108f8 100644 --- a/tests/test_learnware_client/test_all_learnware.py +++ b/tests/test_learnware_client/test_all_learnware.py @@ -30,15 +30,17 @@ class TestAllLearnware(unittest.TestCase): self.client.login(email, token) def test_all_learnware(self): - max_learnware_num = 1000 + max_learnware_num = 2000 semantic_spec = self.client.create_semantic_specification() user_info = BaseUserInfo(semantic_spec=semantic_spec, stat_info={}) result = self.client.search_learnware(user_info, page_size=max_learnware_num) - print(f"result size: {len(result)}") - print(f"key in result: {[key for key in result[0]]}") + + learnware_ids = result["single"]["learnware_ids"] + keys = [key for key in result["single"]["semantic_specifications"][0]] + print(f"result size: {len(learnware_ids)}") + print(f"key in result: {keys}") failed_ids = [] - learnware_ids = [res["learnware_id"] for res in result] with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: for idx in learnware_ids: zip_path = os.path.join(tempdir, f"test_{idx}.zip")