From 87bb9c614c85d755458a9e2a7e01286aa2041a7d Mon Sep 17 00:00:00 2001 From: Gene Date: Sun, 26 Nov 2023 23:41:20 +0800 Subject: [PATCH] [FIX] fix details to fit with BaseUserInfo --- tests/test_learnware_client/test_all_learnware.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/tests/test_learnware_client/test_all_learnware.py b/tests/test_learnware_client/test_all_learnware.py index 2303089..276ac00 100644 --- a/tests/test_learnware_client/test_all_learnware.py +++ b/tests/test_learnware_client/test_all_learnware.py @@ -6,6 +6,7 @@ import tempfile from learnware.client import LearnwareClient from learnware.specification import Specification +from learnware.market import BaseUserInfo class TestAllLearnware(unittest.TestCase): @@ -30,16 +31,9 @@ class TestAllLearnware(unittest.TestCase): def test_all_learnware(self): max_learnware_num = 1000 - semantic_spec = dict() - semantic_spec["Data"] = {"Type": "Class", "Values": []} - semantic_spec["Task"] = {"Type": "Class", "Values": []} - semantic_spec["Library"] = {"Type": "Class", "Values": []} - semantic_spec["Scenario"] = {"Type": "Tag", "Values": []} - semantic_spec["Name"] = {"Type": "String", "Values": ""} - semantic_spec["Description"] = {"Type": "String", "Values": ""} - - specification = Specification(semantic_spec=semantic_spec) - result = self.client.search_learnware(specification, page_size=max_learnware_num) + 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]]}")