def test_print_treenode_with_children(self):
node = TreeNode(data={
"id": "X",
"deprel": "Y",
"test": "data"
},
children=[
TreeNode(data={
"id": "X",
"deprel": "Y",
"test": "data"
},
children=[]),
TreeNode(data={
"id": "X",
"deprel": "Y",
"test": "data"
},
children=[]),
])
result = self._capture_print(print_tree, node)
self.assertEqual(
result,
dedent("""
(deprel:Y) test:data [X]
(deprel:Y) test:data [X]
(deprel:Y) test:data [X]
""").lstrip())
def test_print_simple_treenode(self):
node = TreeNode(data={
"id": "X",
"deprel": "Y",
"test": "data"
},
children=[])
result = self._capture_print(print_tree, node)
self.assertEqual(result, "(deprel:Y) test:data [X]\n")
def test_print_list_of_nodes(self):
node = TreeNode(data={
"id": "X",
"deprel": "Y",
"test": "data"
},
children=[])
nodes = [node, node]
result = self._capture_print(print_tree, nodes)
self.assertEqual(result, "(deprel:Y) test:data [X]\n" * 2)
TreeNode(data=OrderedDict([
('id', 5), ('form', 'jumps'), ('lemma', 'jump'), ('upostag', 'VERB'),
('xpostag', 'VBZ'),
('feats',
OrderedDict([('Mood', 'Ind'), ('Number', 'Sing'), ('Person', '3'),
('Tense', 'Pres'), ('VerbForm', 'Fin')])), ('head', 0),
('deprel', 'root'), ('deps', None), ('misc', None)
]),
children=[
TreeNode(data=OrderedDict([
('id', 4), ('form', 'fox'), ('lemma', 'fox'),
('upostag', 'NOUN'), ('xpostag', 'NN'),
('feats', OrderedDict([('Number', 'Sing')])), ('head', 5),
('deprel', 'nsubj'), ('deps', None), ('misc', None)
]),
children=[
TreeNode(data=OrderedDict([
('id', 1), ('form', 'The'), ('lemma', 'the'),
('upostag', 'DET'), ('xpostag', 'DT'),
('feats',
OrderedDict([('Definite', 'Def'),
('PronType', 'Art')])),
('head', 4), ('deprel', 'det'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 2), ('form', 'quick'),
('lemma', 'quick'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 4), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 3), ('form', 'brown'),
('lemma', 'brown'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 4), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[])
]),
TreeNode(data=OrderedDict([
('id', 9), ('form', 'dog'), ('lemma', 'dog'),
('upostag', 'NOUN'), ('xpostag', 'NN'),
('feats', OrderedDict([('Number', 'Sing')])), ('head', 5),
('deprel', 'nmod'), ('deps', None),
('misc', OrderedDict([('SpaceAfter', 'No')]))
]),
children=[
TreeNode(data=OrderedDict([('id', 6),
('form', 'over'),
('lemma', 'over'),
('upostag', 'ADP'),
('xpostag', 'IN'),
('feats', None),
('head', 9),
('deprel', 'case'),
('deps', None),
('misc', None)]),
children=[]),
TreeNode(data=OrderedDict([
('id', 7), ('form', 'the'), ('lemma', 'the'),
('upostag', 'DET'), ('xpostag', 'DT'),
('feats',
OrderedDict([('Definite', 'Def'),
('PronType', 'Art')])),
('head', 9), ('deprel', 'det'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 8), ('form', 'lazy'),
('lemma', 'lazy'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 9), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[])
]),
TreeNode(data=OrderedDict([('id', 10), ('form', '.'),
('lemma', '.'),
('upostag', 'PUNCT'),
('xpostag', '.'), ('feats', None),
('head', 5), ('deprel', 'punct'),
('deps', None), ('misc', None)]),
children=[])
])
TreeNode(data=OrderedDict([
('id', 5), ('form', 'jumps'), ('lemma', 'jump'), ('upostag', 'VERB'),
('xpostag', 'VBZ'),
('feats',
OrderedDict([('Mood', 'Ind'), ('Number', 'Sing'), ('Person', '3'),
('Tense', 'Pres'), ('VerbForm', 'Fin')])), ('head', 0),
('deprel', 'root'), ('deps', None), ('misc', None)
]),
children=[
TreeNode(data=OrderedDict([
('id', 4), ('form', 'fox'), ('lemma', 'fox'),
('upostag', 'NOUN'), ('xpostag', 'NN'),
('feats', OrderedDict([('Number', 'Sing')])), ('head', 5),
('deprel', 'nsubj'), ('deps', None), ('misc', None)
]),
children=[
TreeNode(data=OrderedDict([
('id', 1), ('form', 'The'), ('lemma', 'the'),
('upostag', 'DET'), ('xpostag', 'DT'),
('feats',
OrderedDict([('Definite', 'Def'),
('PronType', 'Art')])),
('head', 4), ('deprel', 'det'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 2), ('form', 'quick'),
('lemma', 'quick'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 4), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 3), ('form', 'brown'),
('lemma', 'brown'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 4), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[])
]),
TreeNode(data=OrderedDict([
('id', 9), ('form', 'dog'), ('lemma', 'dog'),
('upostag', 'NOUN'), ('xpostag', 'NN'),
('feats', OrderedDict([('Number', 'Sing')])), ('head', 5),
('deprel', 'nmod'), ('deps', None),
('misc', OrderedDict([('SpaceAfter', 'No')]))
]),
children=[
TreeNode(data=OrderedDict([('id', 6),
('form', 'over'),
('lemma', 'over'),
('upostag', 'ADP'),
('xpostag', 'IN'),
('feats', None),
('head', 9),
('deprel', 'case'),
('deps', None),
('misc', None)]),
children=[]),
TreeNode(data=OrderedDict([
('id', 7), ('form', 'the'), ('lemma', 'the'),
('upostag', 'DET'), ('xpostag', 'DT'),
('feats',
OrderedDict([('Definite', 'Def'),
('PronType', 'Art')])),
('head', 9), ('deprel', 'det'),
('deps', None), ('misc', None)
]),
children=[]),
TreeNode(data=OrderedDict([
('id', 8), ('form', 'lazy'),
('lemma', 'lazy'), ('upostag', 'ADJ'),
('xpostag', 'JJ'),
('feats', OrderedDict([('Degree', 'Pos')])),
('head', 9), ('deprel', 'amod'),
('deps', None), ('misc', None)
]),
children=[])
]),
TreeNode(data=OrderedDict([('id', 10), ('form', '.'),
('lemma', '.'),
('upostag', 'PUNCT'),
('xpostag', '.'), ('feats', None),
('head', 5), ('deprel', 'punct'),
('deps', None), ('misc', None)]),
children=[])
])