def setUp(self): g = networkx.DiGraph() self.container = GraphContainer(graph=g) self.tool = GraphNodeSelectionTool(component=self.container) self.container.tools.append(self.tool) node = GraphNodeComponent(position=[0, 0]) self.container.components.append(node)
def __canvas_default(self): """ default setter for _canvas """ if self.graph.is_directed(): container = DAGContainer(style=self.layout) else: container = GraphContainer(style=self.layout) container.tools.append(GraphNodeSelectionTool(component=container)) container.tools.append( GraphNodeHoverTool(component=container, callback=self._on_hover)) container.tools.append(GraphNodeDragTool(component=container)) return container