From e1093f3beb5ee7488cd7ce4bc282c4eaaf77ff73 Mon Sep 17 00:00:00 2001 From: Gene Date: Sun, 29 Oct 2023 17:58:10 +0800 Subject: [PATCH] [MNT] expand testing scope --- tests/test_market/test_easy.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/tests/test_market/test_easy.py b/tests/test_market/test_easy.py index 2aa674c..aa7cad8 100644 --- a/tests/test_market/test_easy.py +++ b/tests/test_market/test_easy.py @@ -26,6 +26,12 @@ user_semantic = { "Scenario": {"Values": ["Education"], "Type": "Tag"}, "Description": {"Values": "", "Type": "String"}, "Name": {"Values": "", "Type": "String"}, + "Output": { + "Dimension": 10, + "Description": { + "0": "the probability of the label is zero", + }, + } } @@ -85,28 +91,36 @@ class TestMarket(unittest.TestCase): self.zip_path_list.append(zip_file) - def test_upload_delete_learnware(self, learnware_num=5, delete=False): + def test_upload_delete_learnware(self, learnware_num=5, delete=True): easy_market = self._init_learnware_market() self.test_prepare_learnware_randomly(learnware_num) + self.learnware_num = learnware_num print("Total Item:", len(easy_market)) + assert len(easy_market) == 0, f"The market should be empty!" for idx, zip_path in enumerate(self.zip_path_list): semantic_spec = copy.deepcopy(user_semantic) semantic_spec["Name"]["Values"] = "learnware_%d" % (idx) semantic_spec["Description"]["Values"] = "test_learnware_number_%d" % (idx) - semantic_spec["Output"] = {"Dimension": 1, "Description": {"0": "The label of the hand-written digit."}} easy_market.add_learnware(zip_path, semantic_spec) print("Total Item:", len(easy_market)) + assert len(easy_market) == self.learnware_num, f"The number of learnwares must be {self.learnware_num}!" + curr_inds = easy_market.get_learnware_ids() print("Available ids After Uploading Learnwares:", curr_inds) + assert len(curr_inds) == self.learnware_num, f"The number of learnwares must be {self.learnware_num}!" if delete: for learnware_id in curr_inds: easy_market.delete_learnware(learnware_id) + self.learnware_num -= 1 + assert len(easy_market) == self.learnware_num, f"The number of learnwares must be {self.learnware_num}!" + curr_inds = easy_market.get_learnware_ids() print("Available ids After Deleting Learnwares:", curr_inds) + assert len(curr_inds) == 0, f"The market should be empty!" return easy_market