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()