diff --git a/tests/test_learnware_client/test_check_learnware.py b/tests/test_learnware_client/test_check_learnware.py new file mode 100644 index 0000000..df77c41 --- /dev/null +++ b/tests/test_learnware_client/test_check_learnware.py @@ -0,0 +1,27 @@ +import os +import unittest +import tempfile + + +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) + self.learnware_id = "00000154" + + def test_check_learnware(self): + with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: + self.zip_path = os.path.join(tempdir, "test.zip") + self.client.download_learnware(self.learnware_id, self.zip_path) + LearnwareClient.check_learnware(self.zip_path) + + +if __name__ == "__main__": + unittest.main() \ No newline at end of file diff --git a/tests/test_learnware_client/test_learnware.py b/tests/test_learnware_client/test_learnware.py deleted file mode 100644 index 4fee31a..0000000 --- a/tests/test_learnware_client/test_learnware.py +++ /dev/null @@ -1,23 +0,0 @@ -import os -import zipfile -import tempfile -from learnware.learnware import get_learnware_from_dirpath -from learnware.test import get_semantic_specification -from learnware.client.container import LearnwaresContainer -from learnware.market import EasyMarket - -if __name__ == "__main__": - semantic_specification = get_semantic_specification() - - zip_path = "rf_tic.zip" - with tempfile.TemporaryDirectory(suffix="learnware") as tempdir: - learnware_dirpath = os.path.join(tempdir, "test") - with zipfile.ZipFile(zip_path, "r") as z_file: - z_file.extractall(learnware_dirpath) - learnware = get_learnware_from_dirpath( - id="test", semantic_spec=semantic_specification, learnware_dirpath=learnware_dirpath - ) - - with LearnwaresContainer(learnware, zip_path) as env_container: - learnware = env_container.get_learnwares_with_container()[0] - print(EasyMarket.check_learnware(learnware))