From 2cb3e96b210548410c3f0a67adf71ae2eb686e8b Mon Sep 17 00:00:00 2001 From: Gene Date: Thu, 26 Oct 2023 19:33:57 +0800 Subject: [PATCH] [MNT] add check learnware with conda yaml --- tests/test_learnware_client/test_check_learnware.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/test_learnware_client/test_check_learnware.py b/tests/test_learnware_client/test_check_learnware.py index aa21543..677e6eb 100644 --- a/tests/test_learnware_client/test_check_learnware.py +++ b/tests/test_learnware_client/test_check_learnware.py @@ -14,12 +14,19 @@ class TestCheckLearnware(unittest.TestCase): self.client = LearnwareClient() self.client.login(email, token) - self.learnware_id = "00000154" - def test_check_learnware(self): + def test_check_learnware_pip(self): + learnware_id = "00000154" 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) + self.client.download_learnware(learnware_id, self.zip_path) + LearnwareClient.check_learnware(self.zip_path) + + def test_check_learnware_conda(self): + learnware_id = "00000148" + with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: + self.zip_path = os.path.join(tempdir, "test.zip") + self.client.download_learnware(learnware_id, self.zip_path) LearnwareClient.check_learnware(self.zip_path)