From d7ae0cc5a250623287ae4da050d623faf438094c Mon Sep 17 00:00:00 2001 From: eneller Date: Tue, 4 Mar 2025 18:41:38 +0100 Subject: [PATCH] test: basic file count --- test/__init__.py | 0 test/test_epub.py | 23 +++++++++++++++++++++++ 2 files changed, 23 insertions(+) create mode 100644 test/__init__.py create mode 100644 test/test_epub.py diff --git a/test/__init__.py b/test/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/test/test_epub.py b/test/test_epub.py new file mode 100644 index 0000000..55f546c --- /dev/null +++ b/test/test_epub.py @@ -0,0 +1,23 @@ +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()