File size: 301 Bytes
6410a28
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import unittest
from app.models.upload import upload_model_to_hf
from app.models.fetch import fetch_models

class TestModels(unittest.TestCase):
    def test_fetch_models(self):
        models = fetch_models()
        self.assertIsInstance(models, list)

if __name__ == "__main__":
    unittest.main()