diff --git a/tests/test_learnware_client/test_check_learnware.py b/tests/test_learnware_client/test_check_learnware.py index 1979026..1e4ed55 100644 --- a/tests/test_learnware_client/test_check_learnware.py +++ b/tests/test_learnware_client/test_check_learnware.py @@ -6,11 +6,19 @@ import tempfile from learnware.client import LearnwareClient + class TestCheckLearnware(unittest.TestCase): def setUp(self): unittest.TestCase.setUpClass() self.client = LearnwareClient() + def test_check_learnware_pip_only_zip(self): + learnware_id = "00000208" + 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) + def test_check_learnware_pip(self): learnware_id = "00000208" with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: @@ -66,8 +74,10 @@ class TestCheckLearnware(unittest.TestCase): semantic_spec = json.load(json_file) LearnwareClient.check_learnware(self.zip_path, semantic_spec) + def suite(): _suite = unittest.TestSuite() + _suite.addTest(TestCheckLearnware("test_check_learnware_pip_only_zip")) _suite.addTest(TestCheckLearnware("test_check_learnware_pip")) _suite.addTest(TestCheckLearnware("test_check_learnware_conda")) _suite.addTest(TestCheckLearnware("test_check_learnware_dependency")) @@ -75,6 +85,7 @@ def suite(): _suite.addTest(TestCheckLearnware("test_check_learnware_text")) return _suite + if __name__ == "__main__": runner = unittest.TextTestRunner() - runner.run(suite()) \ No newline at end of file + runner.run(suite())