diff options
| author | Fabrice Desclaux <fabrice.desclaux@cea.fr> | 2015-01-09 15:34:52 +0100 |
|---|---|---|
| committer | Fabrice Desclaux <fabrice.desclaux@cea.fr> | 2015-01-09 15:34:52 +0100 |
| commit | 48581006f578d7de1a5944c31952faa5a4e947b4 (patch) | |
| tree | e09503cc60f0ce9576a830a2bdc78b7f303961f2 /miasm2/core/graph.py | |
| parent | d421bd8b3dae7a7c1dd81eb2e2a90e500fec8b67 (diff) | |
| download | miasm-48581006f578d7de1a5944c31952faa5a4e947b4.tar.gz miasm-48581006f578d7de1a5944c31952faa5a4e947b4.zip | |
Add docstring/Fix variable name
Diffstat (limited to 'miasm2/core/graph.py')
| -rw-r--r-- | miasm2/core/graph.py | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/miasm2/core/graph.py b/miasm2/core/graph.py index ad159771..d29517cc 100644 --- a/miasm2/core/graph.py +++ b/miasm2/core/graph.py @@ -27,13 +27,16 @@ class DiGraph: self._nodes_to[n] = [] self._nodes_from[n] = [] - def del_node(self, n): - if n in self._nodes: - self._nodes.remove(n) - for pred in self.predecessors(n): - self.del_edge(pred, n) - for succ in self.successors(n): - self.del_edge(n, succ) + def del_node(self, node): + """Delete the @node of the graph; Also delete every edge to/from this + @node""" + + if node in self._nodes: + self._nodes.remove(node) + for pred in self.predecessors(node): + self.del_edge(pred, node) + for succ in self.successors(node): + self.del_edge(node, succ) def add_edge(self, a, b): if not a in self._nodes: |