From d1dd5242d3139b964f7a5485afd5a8ed672c4600 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Wed, 12 Apr 2017 23:19:53 +0100 Subject: Add tests for paragraph transform --- tests/transforms_tests.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) 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,10 +1,25 @@ 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 -- cgit v1.2.3