tests/test_pretrained.py

import unittest


class TestPretrained(unittest.TestCase):
    def test_download(self):
        pass

    def test_generation(self):
        pass


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