def test_indent_tree(self):
requested = ["P7", "P8"]
mgr2 = NodeManager(
{"Start Datetime": datetime.now()}, 10, self.lfl_params, requested, [], self.derived_nodes, {}, {}
)
gr = graph_nodes(mgr2)
gr.node["Raw1"]["active"] = True
gr.node["Raw2"]["active"] = False
gr.node["P4"]["active"] = False
self.assertEqual(
indent_tree(gr, "P7", space="__", delim=" ", label=False),
[" P7", "__ [P4]", "____ Raw1", "____ [Raw2]", "__ P5", "____ Raw3", "____ Raw4", "__ P6", "____ Raw3"],
)
# don't recurse valid parameters...
self.assertEqual(indent_tree(gr, "P5", label=False, recurse_active=False), [])
self.assertEqual(indent_tree(gr, "P4", label=False, recurse_active=False), ["- [P4]", " - [Raw2]"])
self.assertEqual(
indent_tree(gr, "root"),
[
"- root",
" - P7 (DerivedParameterNode)",
" - [P4] (DerivedParameterNode)",
" - Raw1 (HDFNode)",
" - [Raw2] (HDFNode)",
" - P5 (DerivedParameterNode)",
" - Raw3 (HDFNode)",
" - Raw4 (HDFNode)",
" - P6 (DerivedParameterNode)",
" - Raw3 (HDFNode)",
" - P8 (DerivedParameterNode)",
" - Raw5 (HDFNode)",
],
)
def test_indent_tree(self):
requested = ['P7', 'P8']
mgr2 = NodeManager({'Start Datetime': datetime.now()}, 10, self.lfl_params,
requested, [], self.derived_nodes, {}, {})
gr = graph_nodes(mgr2)
gr.node['Raw1']['active'] = True
gr.node['Raw2']['active'] = False
gr.node['P4']['active'] = False
self.assertEqual(
indent_tree(gr, 'P7', space='__', delim=' ', label=False),
[' P7',
'__ [P4]',
'____ Raw1',
'____ [Raw2]',
'__ P5',
'____ Raw3',
'____ Raw4',
'__ P6',
'____ Raw3',
])
# don't recurse valid parameters...
self.assertEqual(
indent_tree(gr, 'P5', label=False, recurse_active=False),
[])
self.assertEqual(
indent_tree(gr, 'P4', label=False, recurse_active=False),
['- [P4]',
' - [Raw2]',
])
self.assertEqual(
indent_tree(gr, 'root'),
['- root',
' - P7 (DerivedParameterNode)',
' - [P4] (DerivedParameterNode)',
' - Raw1 (HDFNode)',
' - [Raw2] (HDFNode)',
' - P5 (DerivedParameterNode)',
' - Raw3 (HDFNode)',
' - Raw4 (HDFNode)',
' - P6 (DerivedParameterNode)',
' - Raw3 (HDFNode)',
' - P8 (DerivedParameterNode)',
' - Raw5 (HDFNode)',
])
def test_indent_tree(self):
requested = ['P7', 'P8']
mgr2 = NodeManager({'Start Datetime': datetime.now()}, 10, self.lfl_params,
requested, [], self.derived_nodes, {}, {})
gr = graph_nodes(mgr2)
gr.node['Raw1']['active'] = True
gr.node['Raw2']['active'] = False
gr.node['P4']['active'] = False
self.assertEqual(
indent_tree(gr, 'P7', space='__', delim=' ', label=False),
[' P7',
'__ [P4]',
'____ Raw1',
'____ [Raw2]',
'__ P5',
'____ Raw3',
'____ Raw4',
'__ P6',
'____ Raw3',
])
# don't recurse valid parameters...
self.assertEqual(
indent_tree(gr, 'P5', label=False, recurse_active=False),
[])
self.assertEqual(
indent_tree(gr, 'P4', label=False, recurse_active=False),
['- [P4]',
' - [Raw2]',
])
self.assertEqual(
indent_tree(gr, 'root'),
['- root',
' - P7 (DerivedParameterNode)',
' - [P4] (DerivedParameterNode)',
' - Raw1 (HDFNode)',
' - [Raw2] (HDFNode)',
' - P5 (DerivedParameterNode)',
' - Raw3 (HDFNode)',
' - Raw4 (HDFNode)',
' - P6 (DerivedParameterNode)',
' - Raw3 (HDFNode)',
' - P8 (DerivedParameterNode)',
' - Raw5 (HDFNode)',
])
