Browse Source

[FIX] fit with client

tags/v0.3.2
GeneLiuXe 2 years ago
parent
commit
47f2bbc7a8
1 changed files with 6 additions and 4 deletions
  1. +6
    -4
      tests/test_learnware_client/test_all_learnware.py

+ 6
- 4
tests/test_learnware_client/test_all_learnware.py View File

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


Loading…
Cancel
Save