1 files changed, 56 insertions, 0 deletions
diff --git a/plugins/render_math/test_render_math.py b/plugins/render_math/test_render_math.py
new file mode 100755
index 0000000..b71f4e7
--- /dev/null
+++ b/plugins/render_math/test_render_math.py
@@ -0,0 +1,56 @@
+# -*- coding: utf-8 -*-
+from os.path import dirname, join
+from tempfile import TemporaryDirectory
+
+from pelican import Pelican
+from pelican.generators import ArticlesGenerator
+from pelican.settings import configure_settings
+from pelican.tests.support import get_settings, unittest
+from pelican.writers import Writer
+
+from .math import pelican_init, process_rst_and_summaries
+
+
+CUR_DIR = dirname(__file__)
+
+
+class RenderMathTest(unittest.TestCase):
+    def test_ok_on_shared_test_data(self):
+        settings = get_settings(filenames={})
+        settings['PATH'] = join(CUR_DIR, '..', 'test_data')
+        pelican_init(PelicanMock(settings))
+        with TemporaryDirectory() as tmpdirname:
+            generator = _build_article_generator(settings, tmpdirname)
+            process_rst_and_summaries([generator])
+    def test_ok_on_custom_data(self):
+        settings = get_settings(filenames={})
+        settings['PATH'] = join(CUR_DIR, 'test_data')
+        settings['PLUGINS'] = ['pelican-ipynb.markup']  # to also parse .ipynb files
+        configure_settings(settings)
+        pelican_mock = PelicanMock(settings)
+        pelican_init(pelican_mock)
+        Pelican.init_plugins(pelican_mock)
+        with TemporaryDirectory() as tmpdirname:
+            generator = _build_article_generator(settings, tmpdirname)
+            process_rst_and_summaries([generator])
+            for article in generator.articles:
+                if article.source_path.endswith('.rst'):
+                    self.assertIn('mathjaxscript_pelican', article.content)
+            generator.generate_output(Writer(tmpdirname, settings=settings))
+
+
+def _build_article_generator(settings, output_path):
+    context = settings.copy()
+    context['generated_content'] = dict()
+    context['static_links'] = set()
+    article_generator = ArticlesGenerator(
+        context=context, settings=settings,
+        path=settings['PATH'], theme=settings['THEME'], output_path=output_path)
+    article_generator.generate_context()
+    return article_generator
+
+class PelicanMock:
+    'A dummy class exposing the only attributes needed'
+    def __init__(self, settings):
+        self.plugins = []
+        self.settings = settings