From ec8ffb1f2790193a7d4c313b4d3e7d4873d0e999 Mon Sep 17 00:00:00 2001 From: Gene Date: Sun, 19 Nov 2023 15:44:57 +0800 Subject: [PATCH] [MNT] modify details in client tests --- .../test_all_learnware.py | 18 +++++++++++++++--- .../test_check_learnware.py | 4 ---- tests/test_learnware_client/test_load_conda.py | 4 ---- .../test_learnware_client/test_load_docker.py | 4 ---- tests/test_learnware_client/test_upload.py | 16 ++++++++++++++-- 5 files changed, 29 insertions(+), 17 deletions(-) diff --git a/tests/test_learnware_client/test_all_learnware.py b/tests/test_learnware_client/test_all_learnware.py index c91222d..37b9046 100644 --- a/tests/test_learnware_client/test_all_learnware.py +++ b/tests/test_learnware_client/test_all_learnware.py @@ -1,4 +1,5 @@ import os +import json import unittest import tempfile @@ -9,9 +10,20 @@ from learnware.specification import Specification class TestAllLearnware(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() - email = "liujd@lamda.nju.edu.cn" - token = "f7e647146a314c6e8b4e2e1079c4bca4" - + dir_path = os.path.dirname(__file__) + config_path = os.path.join(dir_path, "config.json") + if not os.path.exists(config_path): + data = {"email": None, "token": None} + with open(config_path, "w") as file: + json.dump(data, file) + + with open(config_path, "r") as file: + data = json.load(file) + email = data["email"] + token = data["token"] + + if email is None or token is None: + raise ValueError("Please set email and token in config.json.") self.client = LearnwareClient() self.client.login(email, token) diff --git a/tests/test_learnware_client/test_check_learnware.py b/tests/test_learnware_client/test_check_learnware.py index a5c5297..62f4c5b 100644 --- a/tests/test_learnware_client/test_check_learnware.py +++ b/tests/test_learnware_client/test_check_learnware.py @@ -9,11 +9,7 @@ from learnware.client import LearnwareClient class TestCheckLearnware(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() - email = "liujd@lamda.nju.edu.cn" - token = "f7e647146a314c6e8b4e2e1079c4bca4" - self.client = LearnwareClient() - self.client.login(email, token) def test_check_learnware_pip(self): learnware_id = "00000154" diff --git a/tests/test_learnware_client/test_load_conda.py b/tests/test_learnware_client/test_load_conda.py index 7fe951c..4394348 100644 --- a/tests/test_learnware_client/test_load_conda.py +++ b/tests/test_learnware_client/test_load_conda.py @@ -13,11 +13,7 @@ from learnware.reuse import AveragingReuser class TestLearnwareLoad(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() - email = "liujd@lamda.nju.edu.cn" - token = "f7e647146a314c6e8b4e2e1079c4bca4" - self.client = LearnwareClient() - self.client.login(email, token) root = os.path.dirname(__file__) self.learnware_ids = ["00000084", "00000154", "00000155"] diff --git a/tests/test_learnware_client/test_load_docker.py b/tests/test_learnware_client/test_load_docker.py index b64ffed..775405c 100644 --- a/tests/test_learnware_client/test_load_docker.py +++ b/tests/test_learnware_client/test_load_docker.py @@ -13,11 +13,7 @@ from learnware.reuse import AveragingReuser class TestLearnwareLoad(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() - email = "liujd@lamda.nju.edu.cn" - token = "f7e647146a314c6e8b4e2e1079c4bca4" - self.client = LearnwareClient() - self.client.login(email, token) root = os.path.dirname(__file__) self.learnware_ids = ["00000084", "00000154", "00000155"] diff --git a/tests/test_learnware_client/test_upload.py b/tests/test_learnware_client/test_upload.py index 7bd3128..a38f938 100644 --- a/tests/test_learnware_client/test_upload.py +++ b/tests/test_learnware_client/test_upload.py @@ -1,4 +1,5 @@ import os +import json import unittest import tempfile @@ -8,9 +9,20 @@ from learnware.client import LearnwareClient class TestAllLearnware(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() - email = "liujd@lamda.nju.edu.cn" - token = "f7e647146a314c6e8b4e2e1079c4bca4" + dir_path = os.path.dirname(__file__) + config_path = os.path.join(dir_path, "config.json") + if not os.path.exists(config_path): + data = {"email": None, "token": None} + with open(config_path, "w") as file: + json.dump(data, file) + with open(config_path, "r") as file: + data = json.load(file) + email = data["email"] + token = data["token"] + + if email is None or token is None: + raise ValueError("Please set email and token in config.json.") self.client = LearnwareClient() self.client.login(email, token)