summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMridul Seth <seth.mridul@gmail.com>2015-06-16 20:39:51 +0530
committerMridul Seth <seth.mridul@gmail.com>2015-06-16 20:39:51 +0530
commit902e221ae7b9e8c32489afcfdc815c2839342249 (patch)
treed671eb1b11bd00c24ce19ba762fb677c8b6841e6
parentb433dcc99c5728fb16e599a7153fbb1b834d9167 (diff)
downloadnetworkx-902e221ae7b9e8c32489afcfdc815c2839342249.tar.gz
Update changes to G.successors()
-rw-r--r--networkx/classes/tests/test_digraph_historical.py3
-rw-r--r--networkx/generators/directed.py2
2 files changed, 1 insertions, 4 deletions
diff --git a/networkx/classes/tests/test_digraph_historical.py b/networkx/classes/tests/test_digraph_historical.py
index 63f193b9..141b1bfe 100644
--- a/networkx/classes/tests/test_digraph_historical.py
+++ b/networkx/classes/tests/test_digraph_historical.py
@@ -59,9 +59,6 @@ class TestDiGraphHistorical(HistoricalTests):
assert_equal(sorted(G.neighbors('C')),['D'])
assert_equal(sorted(G['C']),['D'])
assert_equal(sorted(G.neighbors('A')),['B', 'C'])
- assert_equal(sorted(G.neighbors('A')),['B', 'C'])
- assert_equal(sorted(G.neighbors('C')),['D'])
- assert_equal(sorted(G.neighbors('A')),['B', 'C'])
assert_raises(nx.NetworkXError,G.neighbors,'j')
assert_raises(nx.NetworkXError,G.neighbors,'j')
diff --git a/networkx/generators/directed.py b/networkx/generators/directed.py
index ee229ba1..1ff2aa24 100644
--- a/networkx/generators/directed.py
+++ b/networkx/generators/directed.py
@@ -144,7 +144,7 @@ def gnr_graph(n, p, create_using=None, seed=None):
for source in range(1,n):
target=random.randrange(0,source)
if random.random() < p and target !=0:
- target = list(G.successors(target))[0]
+ target = next(G.successors(target))
G.add_edge(source,target)
return G