def print_node_tree(cls, node, max_depth, follow_links=()): """Top-level function for printing node tree.""" from ete3 import Tree from aiida.cmdline.utils.common import get_node_summary echo.echo(get_node_summary(node)) tree_string = '({});'.format( cls._build_tree(node, max_depth=max_depth, follow_links=follow_links)) tmp = Tree(tree_string, format=1) echo.echo(tmp.get_ascii(show_internal=True))
def print_node_tree(cls, node, max_depth, follow_links=()): """Top-level function for printing node tree.""" import warnings from aiida.common.warnings import AiidaDeprecationWarning warnings.warn( 'class is deprecated and will be removed in `aiida-core==2.0.0`.', AiidaDeprecationWarning) # pylint: disable=no-member from ete3 import Tree from aiida.cmdline.utils.common import get_node_summary from aiida.cmdline.utils import echo echo.echo(get_node_summary(node)) tree_string = f'({cls._build_tree(node, max_depth=max_depth, follow_links=follow_links)});' tmp = Tree(tree_string, format=1) echo.echo(tmp.get_ascii(show_internal=True))
def test_get_node_summary(self): """Test the `get_node_summary` utility.""" from aiida.cmdline.utils.common import get_node_summary computer_label = self.computer.name # pylint: disable=no-member code = orm.Code( input_plugin_name='arithmetic.add', remote_computer_exec=[self.computer, '/remote/abs/path'], ) code.store() node = orm.CalculationNode() node.computer = self.computer node.add_incoming(code, link_type=LinkType.INPUT_CALC, link_label='code') node.store() summary = get_node_summary(node) self.assertIn(node.uuid, summary) self.assertIn(computer_label, summary)