2 files changed, 32 insertions, 0 deletions
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/test_pymdvar.py b/tests/test_pymdvar.py
new file mode 100644
index 0000000..2e305ec
--- /dev/null
+++ b/tests/test_pymdvar.py
@@ -0,0 +1,32 @@
+import pytest
+from markdown import markdown
+from pymdvar import VariableExtension
+
+
+def test_empty_input():
+    in_str = ''
+    out_str = markdown(in_str, extensions=[VariableExtension()])
+    print(in_str)
+    print(out_str)
+    assert in_str == out_str
+
+
+@pytest.mark.parametrize('in_str, exp_str', [
+    ('foo bar', '<p>foo bar</p>'),
+    ('foo *test* bar', '<p>foo <em>test</em> bar</p>'),
+    ('foo **test** bar', '<p>foo <strong>test</strong> bar</p>')
+])
+def test_non_replacements(in_str, exp_str):
+    out_str = markdown(in_str, extensions=[VariableExtension()])
+    assert out_str == exp_str
+
+
+@pytest.mark.parametrize('in_str, exp_str', [
+    ('foo ${test} bar', '<p>foo value bar</p>'),
+    ('foo *${test}* bar', '<p>foo <em>value</em> bar</p>'),
+    ('foo **${test}** bar', '<p>foo <strong>value</strong> bar</p>')
+])
+def test_simple_replacements(in_str, exp_str):
+    out_str = markdown(in_str, extensions=[VariableExtension(variables={'test':'value'})])
+    assert out_str == exp_str
+