def test_tree_plot_custom_edge(self):
a = ytree.load(AHF, hubble_constant=0.7)
p = ytree.TreePlot(a[0],
dot_kwargs={'rankdir': "BT"},
node_function=my_node,
edge_function=my_edge)
p.save('tree_custom_edge.png')
def test_edge_function(self):
def my_func(desc, anc):
return {"color": "red"}
a = ytree.load(CT)
p = ytree.TreePlot(a[0], edge_function=my_func)
p.save()
def test_tree_plot(self):
a = ytree.load(AHF, hubble_constant=0.7)
p = ytree.TreePlot(a[0],
dot_kwargs={
'rankdir': 'LR',
'size': '"12,4"'
})
p.save('tree.png')
def test_node_function(self):
def my_func(halo):
label = f"{halo['uid']}"
return {"label": label}
a = ytree.load(CT)
p = ytree.TreePlot(a[0], node_function=my_func)
p.save()
def test_tree_plot_small(self):
a = ytree.load(AHF, hubble_constant=0.7)
p = ytree.TreePlot(a[0],
dot_kwargs={
'rankdir': 'LR',
'size': '"12,4"'
})
p.min_mass_ratio = 0.01
p.save('tree_small.png')
def test_non_defaults(self):
attrs = {
'size_field': 'virial_radius',
'size_log': False,
'min_mass': 1e14,
'min_mass_ratio': 0.1
}
a = ytree.load(CT)
for attr, val in attrs.items():
p = ytree.TreePlot(a[0])
setattr(p, attr, val)
p.save()
def test_edge_function_bad(self):
a = ytree.load(CT)
with self.assertRaises(RuntimeError):
ytree.TreePlot(a[0], edge_function='notafunc')
def test_dot_kwargs(self):
a = ytree.load(CT)
p = ytree.TreePlot(a[0], dot_kwargs={'dpi': 200})
p.save()
def test_save(self):
a = ytree.load(CT)
p = ytree.TreePlot(a[0])
p.save('tree.png')
def test_default_plot(self):
a = ytree.load(CT)
p = ytree.TreePlot(a[0])
p.save()
