yonikremer commited on
Commit
2f94880
1 Parent(s): b7eab4d

add more test cases to TestIsDownloaded

Browse files
Files changed (1) hide show
  1. 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")