|
|
|
@@ -35,6 +35,11 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
license=Licenses.APACHE_V2, |
|
|
|
chinese_name=TEST_MODEL_CHINESE_NAME, |
|
|
|
) |
|
|
|
|
|
|
|
def tearDown(self): |
|
|
|
self.api.delete_model(model_id=self.model_id) |
|
|
|
|
|
|
|
def prepare_case(self): |
|
|
|
temporary_dir = tempfile.mkdtemp() |
|
|
|
self.model_dir = os.path.join(temporary_dir, self.model_name) |
|
|
|
repo = Repository(self.model_dir, clone_from=self.model_id) |
|
|
|
@@ -42,9 +47,6 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
% os.path.join(self.model_dir, download_model_file_name)) |
|
|
|
repo.push('add model') |
|
|
|
|
|
|
|
def tearDown(self): |
|
|
|
self.api.delete_model(model_id=self.model_id) |
|
|
|
|
|
|
|
def test_model_repo_creation(self): |
|
|
|
# change to proper model names before use |
|
|
|
try: |
|
|
|
@@ -57,6 +59,7 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
raise |
|
|
|
|
|
|
|
def test_download_single_file(self): |
|
|
|
self.prepare_case() |
|
|
|
downloaded_file = model_file_download( |
|
|
|
model_id=self.model_id, file_path=download_model_file_name) |
|
|
|
assert os.path.exists(downloaded_file) |
|
|
|
@@ -68,6 +71,7 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
assert mdtime1 == mdtime2 |
|
|
|
|
|
|
|
def test_snapshot_download(self): |
|
|
|
self.prepare_case() |
|
|
|
snapshot_path = snapshot_download(model_id=self.model_id) |
|
|
|
downloaded_file_path = os.path.join(snapshot_path, |
|
|
|
download_model_file_name) |
|
|
|
@@ -82,6 +86,7 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
file_path=download_model_file_name) # not add counter |
|
|
|
|
|
|
|
def test_download_public_without_login(self): |
|
|
|
self.prepare_case() |
|
|
|
rmtree(ModelScopeConfig.path_credential) |
|
|
|
snapshot_path = snapshot_download(model_id=self.model_id) |
|
|
|
downloaded_file_path = os.path.join(snapshot_path, |
|
|
|
@@ -96,6 +101,7 @@ class HubOperationTest(unittest.TestCase): |
|
|
|
self.api.login(TEST_ACCESS_TOKEN1) |
|
|
|
|
|
|
|
def test_snapshot_delete_download_cache_file(self): |
|
|
|
self.prepare_case() |
|
|
|
snapshot_path = snapshot_download(model_id=self.model_id) |
|
|
|
downloaded_file_path = os.path.join(snapshot_path, |
|
|
|
download_model_file_name) |
|
|
|
|