Browse Source

[MNT] add check only with zip

tags/v0.3.2
Gene 1 year ago
parent
commit
2257389ea4
1 changed files with 12 additions and 1 deletions
  1. +12
    -1
      tests/test_learnware_client/test_check_learnware.py

+ 12
- 1
tests/test_learnware_client/test_check_learnware.py View File

@@ -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())
runner.run(suite())

Loading…
Cancel
Save