Spaces:
Sleeping
Sleeping
File size: 463 Bytes
83e9505 50310f8 83e9505 384edc4 83e9505 36cbecb 50310f8 8287cb2 36cbecb 83e9505 384edc4 8287cb2 36cbecb 83e9505 384edc4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import unittest
from yume import Yume, Config
class TestPretrained(unittest.TestCase):
def __init__(self, methodName: str = "runTest") -> None:
super().__init__(methodName)
self.config = Config()
self.yume = Yume(config=self.config)
def test_download(self):
self.yume.load_pretrained()
pass
def test_generation(self):
self.yume.sample()
pass
if __name__ == "__main__":
unittest.main()
|