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()