g2h: Add unit tests and restructure class layout

Instead of having a large generator class use individual pages

1 files changed, 19 insertions, 0 deletions

diff --git a/tests/page_test.py b/tests/page_test.py
new file mode 100644
index 0000000..1008a8d
--- /dev/null
+++ b/tests/page_test.py
@@ -0,0 +1,19 @@
+from g2h.page import Page
+from jinja2.exceptions import TemplateNotFound
+import pytest
+
+def test_init():
+    page = Page('overview.md', 'dir', 'file', 'name')
+    assert(page.name == 'name')
+    assert(page.out_file_path == 'dir/file')
+    assert(page.template.name == 'overview.md')
+
+def test_wrong_template():
+    with pytest.raises(TemplateNotFound):
+        Page('not existant.md', 'dir', 'file', 'name')
+
+def test_no_context():
+    page = Page('overview.md', 'dir', 'file', 'name')
+    with pytest.raises(NotImplementedError):
+        page.context()
+