24 lines
684 B
Python
24 lines
684 B
Python
from tqdm import tqdm
|
|
|
|
from src.epub2go.convert import GBConvert, get_all_books
|
|
|
|
import unittest
|
|
|
|
# run using `python -m unittest test/test_epub.py`
|
|
class TestEpub(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.schiller_raeuber = GBConvert('https://www.projekt-gutenberg.org/schiller/raeuber/')
|
|
cls.anzengru_allersee = GBConvert('https://www.projekt-gutenberg.org/anzengru/allersee/')
|
|
|
|
def test_schiller_raeuber_toc(self):
|
|
self.assertEqual(len(self.schiller_raeuber.toc), 7)
|
|
|
|
def test_anzengru_allersee_toc(self):
|
|
self.assertEqual(len(self.anzengru_allersee.toc), 1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|