blob: e4f5360eb392b95990844285cc6f1de256d887d0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from nose.tools import assert_equal, istest
from mammoth import documents
from mammoth.transforms import get_descendants, get_descendants_of_type
@istest
class GetDescendantsTests(object):
@istest
def returns_nothing_if_element_type_has_no_children(self):
assert_equal([], get_descendants(documents.tab()))
@istest
def returns_nothing_if_element_has_empty_children(self):
assert_equal([], get_descendants(documents.paragraph(children=[])))
@istest
def includes_children(self):
children = [documents.text("child 1"), documents.text("child 2")]
element = documents.paragraph(children=children)
assert_equal(children, get_descendants(element))
@istest
def includes_indirect_descendants(self):
grandchild = documents.text("grandchild")
child = documents.run(children=[grandchild])
element = documents.paragraph(children=[child])
assert_equal([grandchild, child], get_descendants(element))
@istest
class GetDescendantsOfTypeTests(object):
@istest
def filters_descendants_to_type(self):
tab = documents.tab()
run = documents.run(children=[])
element = documents.paragraph(children=[tab, run])
assert_equal([run], get_descendants_of_type(element, documents.Run))
|
