From 12e3e2aa7bc80bbbb81eaaa31f719ea4d7ec28f6 Mon Sep 17 00:00:00 2001 From: bxdd Date: Sun, 15 Oct 2023 15:26:21 +0800 Subject: [PATCH] del duplicate code --- tests/test_client/test_reuse.py | 42 ------------------- .../test_learnware.py | 3 -- .../test_load.py | 0 3 files changed, 45 deletions(-) delete mode 100644 tests/test_client/test_reuse.py rename tests/{test_client => test_learnware_client}/test_learnware.py (96%) rename tests/{test_client => test_learnware_client}/test_load.py (100%) diff --git a/tests/test_client/test_reuse.py b/tests/test_client/test_reuse.py deleted file mode 100644 index 5e84f5d..0000000 --- a/tests/test_client/test_reuse.py +++ /dev/null @@ -1,42 +0,0 @@ -import zipfile -import numpy as np - -from learnware.learnware import get_learnware_from_dirpath, Learnware -from learnware.market import EasyMarket -from learnware.client.container import ModelEnvContainer, LearnwaresContainer -from learnware.learnware.reuse import AveragingReuser - -if __name__ == "__main__": - semantic_specification = dict() - semantic_specification["Data"] = {"Type": "Class", "Values": ["Text"]} - semantic_specification["Task"] = {"Type": "Class", "Values": ["Ranking"]} - semantic_specification["Library"] = {"Type": "Class", "Values": ["Scikit-learn"]} - semantic_specification["Scenario"] = {"Type": "Tag", "Values": "Financial"} - semantic_specification["Name"] = {"Type": "String", "Values": "test"} - semantic_specification["Description"] = {"Type": "String", "Values": "test"} - - zip_paths = [ - "/home/bixd/workspace/learnware/Learnware/tests/test_learnware_client/rf_tic.zip", - "/home/bixd/workspace/learnware/Learnware/tests/test_learnware_client/svc_tic.zip", - ] - dir_paths = [ - "/home/bixd/workspace/learnware/Learnware/tests/test_learnware_client/rf_tic", - "/home/bixd/workspace/learnware/Learnware/tests/test_learnware_client/svc_tic", - ] - - learnware_list = [] - for id, (zip_path, dir_path) in enumerate(zip(zip_paths, dir_paths)): - with zipfile.ZipFile(zip_path, "r") as z_file: - z_file.extractall(dir_path) - - learnware = get_learnware_from_dirpath(f"test_id{id}", semantic_specification, dir_path) - learnware_list.append(learnware) - - with LearnwaresContainer(learnware_list, zip_paths) as env_container: - - learnware_list = env_container.get_learnware_list_with_container() - reuser = AveragingReuser(learnware_list, mode="vote_by_label") - input_array = np.random.randint(0, 3, size=(20, 9)) - print(reuser.predict(input_array).argmax(axis=1)) - for id, ind_learner in enumerate(learnware_list): - print(f"learner_{id}", reuser.predict(input_array).argmax(axis=1)) diff --git a/tests/test_client/test_learnware.py b/tests/test_learnware_client/test_learnware.py similarity index 96% rename from tests/test_client/test_learnware.py rename to tests/test_learnware_client/test_learnware.py index 1e27203..2cfff1d 100644 --- a/tests/test_client/test_learnware.py +++ b/tests/test_learnware_client/test_learnware.py @@ -1,6 +1,3 @@ -import os - -import learnware from learnware.client.learnware_client import LearnwareClient diff --git a/tests/test_client/test_load.py b/tests/test_learnware_client/test_load.py similarity index 100% rename from tests/test_client/test_load.py rename to tests/test_learnware_client/test_load.py