yonikremer
commited on
Commit
•
2f94880
1
Parent(s):
b7eab4d
add more test cases to TestIsDownloaded
Browse files- test_is_downloaded.py +2 -3
test_is_downloaded.py
CHANGED
@@ -1,7 +1,4 @@
|
|
1 |
# test_hanlde_form_submit.py - Generated by CodiumAI
|
2 |
-
import os
|
3 |
-
|
4 |
-
import pytest
|
5 |
|
6 |
from hanlde_form_submit import is_downloaded
|
7 |
|
@@ -34,6 +31,8 @@ Test Plan:
|
|
34 |
class TestIsDownloaded:
|
35 |
def test_is_downloaded_exists(self):
|
36 |
assert is_downloaded("gpt2")
|
|
|
|
|
37 |
|
38 |
def test_is_downloaded_not_exists(self):
|
39 |
assert not is_downloaded("non-existent-model")
|
|
|
1 |
# test_hanlde_form_submit.py - Generated by CodiumAI
|
|
|
|
|
|
|
2 |
|
3 |
from hanlde_form_submit import is_downloaded
|
4 |
|
|
|
31 |
class TestIsDownloaded:
|
32 |
def test_is_downloaded_exists(self):
|
33 |
assert is_downloaded("gpt2")
|
34 |
+
assert is_downloaded("facebook/opt-iml-max-1.3b")
|
35 |
+
assert is_downloaded("facebook/opt-iml-max-30b")
|
36 |
|
37 |
def test_is_downloaded_not_exists(self):
|
38 |
assert not is_downloaded("non-existent-model")
|