diff options
Diffstat (limited to 'networkx/algorithms/tests/test_core.py')
-rw-r--r-- | networkx/algorithms/tests/test_core.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/networkx/algorithms/tests/test_core.py b/networkx/algorithms/tests/test_core.py index 9caabfc1..ae546a38 100644 --- a/networkx/algorithms/tests/test_core.py +++ b/networkx/algorithms/tests/test_core.py @@ -54,9 +54,7 @@ class TestCore: def test_find_cores(self): core = nx.find_cores(self.G) - nodes_by_core = [ - sorted([n for n in core if core[n] == val]) for val in range(4) - ] + nodes_by_core = [sorted(n for n in core if core[n] == val) for val in range(4)] assert nodes_equal(nodes_by_core[0], [21]) assert nodes_equal(nodes_by_core[1], [17, 18, 19, 20]) assert nodes_equal(nodes_by_core[2], [9, 10, 11, 12, 13, 14, 15, 16]) @@ -68,9 +66,7 @@ class TestCore: def test_find_cores2(self): core = nx.find_cores(self.H) - nodes_by_core = [ - sorted([n for n in core if core[n] == val]) for val in range(3) - ] + nodes_by_core = [sorted(n for n in core if core[n] == val) for val in range(3)] assert nodes_equal(nodes_by_core[0], [0]) assert nodes_equal(nodes_by_core[1], [1, 3]) assert nodes_equal(nodes_by_core[2], [2, 4, 5, 6]) @@ -169,7 +165,7 @@ class TestCore: def test_onion_layers(self): layers = nx.onion_layers(self.G) nodes_by_layer = [ - sorted([n for n in layers if layers[n] == val]) for val in range(1, 7) + sorted(n for n in layers if layers[n] == val) for val in range(1, 7) ] assert nodes_equal(nodes_by_layer[0], [21]) assert nodes_equal(nodes_by_layer[1], [17, 18, 19, 20]) |