Add tests for paragraph transform

1 files changed, 16 insertions, 1 deletions

diff --git a/tests/transforms_tests.py b/tests/transforms_tests.py
index 0f0a05d..2c1678f 100644
--- a/tests/transforms_tests.py
+++ b/tests/transforms_tests.py
@@ -1,11 +1,26 @@
 import cobble
 from nose.tools import assert_equal, istest
 
-from mammoth import documents
+from mammoth import documents, transforms
 from mammoth.transforms import get_descendants, get_descendants_of_type, _each_element
 
 
 @istest
+class ParagraphTests(object):
+    @istest
+    def paragraph_is_transformed(self):
+        paragraph = documents.paragraph(children=[])
+        result = transforms.paragraph(lambda _: documents.tab())(paragraph)
+        assert_equal(documents.tab(), result)
+        
+    @istest
+    def non_paragraph_elements_are_not_transformed(self):
+        run = documents.run(children=[])
+        result = transforms.paragraph(lambda _: documents.tab())(run)
+        assert_equal(documents.run(children=[]), result)
+
+
+@istest
 class EachElementTests(object):
     @istest
     def all_descendants_are_transformed(self):