summaryrefslogtreecommitdiff
path: root/lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py
diff options
context:
space:
mode:
authorJeff Vander Stoep <jeffv@google.com>2015-06-18 07:52:02 -0700
committerJeff Vander Stoep <jeffv@google.com>2015-06-18 07:52:02 -0700
commitfe63c8d5e6e841f474eec96e2dd38d8fa97a17f8 (patch)
tree57b5a1be2dcd01bc500525ac831f3e3421f4dae5 /lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py
parent5de4f653a1cff7d7acbbb933711537ef47e2723c (diff)
download2.7.5-fe63c8d5e6e841f474eec96e2dd38d8fa97a17f8.tar.gz
SELinux: python modules for selinux tools
<tool(s)>: <required python modules> audit2allow audit2why: sepolgen selinux sesearch: setools setoolsgui Change-Id: Ifc19252d8d8e178b86c51fe1f54e162a61ffc0dd
Diffstat (limited to 'lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py')
-rw-r--r--lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py327
1 files changed, 327 insertions, 0 deletions
diff --git a/lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py b/lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py
new file mode 100644
index 0000000..e8c2543
--- /dev/null
+++ b/lib/python2.7/site-packages/setoolsgui/networkx/classes/tests/test_multidigraph.py
@@ -0,0 +1,327 @@
+#!/usr/bin/env python
+from nose.tools import *
+import networkx
+from test_multigraph import BaseMultiGraphTester, TestMultiGraph
+
+class BaseMultiDiGraphTester(BaseMultiGraphTester):
+ def test_edges(self):
+ G=self.K3
+ assert_equal(sorted(G.edges()),[(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.edges(0)),[(0,1),(0,2)])
+ assert_raises((KeyError,networkx.NetworkXError), G.edges,-1)
+
+ def test_edges_data(self):
+ G=self.K3
+ assert_equal(sorted(G.edges(data=True)),
+ [(0,1,{}),(0,2,{}),(1,0,{}),(1,2,{}),(2,0,{}),(2,1,{})])
+ assert_equal(sorted(G.edges(0,data=True)),[(0,1,{}),(0,2,{})])
+ assert_raises((KeyError,networkx.NetworkXError), G.neighbors,-1)
+
+
+ def test_edges_iter(self):
+ G=self.K3
+ assert_equal(sorted(G.edges_iter()),
+ [(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.edges_iter(0)),[(0,1),(0,2)])
+ G.add_edge(0,1)
+ assert_equal(sorted(G.edges_iter()),
+ [(0,1),(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+
+ def test_out_edges(self):
+ G=self.K3
+ assert_equal(sorted(G.out_edges()),
+ [(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.out_edges(0)),[(0,1),(0,2)])
+ assert_raises((KeyError,networkx.NetworkXError), G.out_edges,-1)
+ assert_equal(sorted(G.out_edges(0,keys=True)),[(0,1,0),(0,2,0)])
+
+ def test_out_edges_iter(self):
+ G=self.K3
+ assert_equal(sorted(G.out_edges_iter()),
+ [(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.out_edges_iter(0)),[(0,1),(0,2)])
+ G.add_edge(0,1,2)
+ assert_equal(sorted(G.out_edges_iter()),
+ [(0,1),(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+
+ def test_in_edges(self):
+ G=self.K3
+ assert_equal(sorted(G.in_edges()),
+ [(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.in_edges(0)),[(1,0),(2,0)])
+ assert_raises((KeyError,networkx.NetworkXError), G.in_edges,-1)
+ G.add_edge(0,1,2)
+ assert_equal(sorted(G.in_edges()),
+ [(0,1),(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.in_edges(0,keys=True)),[(1,0,0),(2,0,0)])
+
+ def test_in_edges_iter(self):
+ G=self.K3
+ assert_equal(sorted(G.in_edges_iter()),
+ [(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+ assert_equal(sorted(G.in_edges_iter(0)),[(1,0),(2,0)])
+ G.add_edge(0,1,2)
+ assert_equal(sorted(G.in_edges_iter()),
+ [(0,1),(0,1),(0,2),(1,0),(1,2),(2,0),(2,1)])
+
+ assert_equal(sorted(G.in_edges_iter(data=True,keys=False)),
+ [(0,1,{}),(0,1,{}),(0,2,{}),(1,0,{}),(1,2,{}),
+ (2,0,{}),(2,1,{})])
+
+
+ def is_shallow(self,H,G):
+ # graph
+ assert_equal(G.graph['foo'],H.graph['foo'])
+ G.graph['foo'].append(1)
+ assert_equal(G.graph['foo'],H.graph['foo'])
+ # node
+ assert_equal(G.node[0]['foo'],H.node[0]['foo'])
+ G.node[0]['foo'].append(1)
+ assert_equal(G.node[0]['foo'],H.node[0]['foo'])
+ # edge
+ assert_equal(G[1][2][0]['foo'],H[1][2][0]['foo'])
+ G[1][2][0]['foo'].append(1)
+ assert_equal(G[1][2][0]['foo'],H[1][2][0]['foo'])
+
+ def is_deep(self,H,G):
+ # graph
+ assert_equal(G.graph['foo'],H.graph['foo'])
+ G.graph['foo'].append(1)
+ assert_not_equal(G.graph['foo'],H.graph['foo'])
+ # node
+ assert_equal(G.node[0]['foo'],H.node[0]['foo'])
+ G.node[0]['foo'].append(1)
+ assert_not_equal(G.node[0]['foo'],H.node[0]['foo'])
+ # edge
+ assert_equal(G[1][2][0]['foo'],H[1][2][0]['foo'])
+ G[1][2][0]['foo'].append(1)
+ assert_not_equal(G[1][2][0]['foo'],H[1][2][0]['foo'])
+
+ def test_to_undirected(self):
+ # MultiDiGraph -> MultiGraph changes number of edges so it is
+ # not a copy operation... use is_shallow, not is_shallow_copy
+ G=self.K3
+ self.add_attributes(G)
+ H=networkx.MultiGraph(G)
+ self.is_shallow(H,G)
+ H=G.to_undirected()
+ self.is_deep(H,G)
+
+ def test_has_successor(self):
+ G=self.K3
+ assert_equal(G.has_successor(0,1),True)
+ assert_equal(G.has_successor(0,-1),False)
+
+ def test_successors(self):
+ G=self.K3
+ assert_equal(sorted(G.successors(0)),[1,2])
+ assert_raises((KeyError,networkx.NetworkXError), G.successors,-1)
+
+ def test_successors_iter(self):
+ G=self.K3
+ assert_equal(sorted(G.successors_iter(0)),[1,2])
+ assert_raises((KeyError,networkx.NetworkXError), G.successors_iter,-1)
+
+ def test_has_predecessor(self):
+ G=self.K3
+ assert_equal(G.has_predecessor(0,1),True)
+ assert_equal(G.has_predecessor(0,-1),False)
+
+ def test_predecessors(self):
+ G=self.K3
+ assert_equal(sorted(G.predecessors(0)),[1,2])
+ assert_raises((KeyError,networkx.NetworkXError), G.predecessors,-1)
+
+ def test_predecessors_iter(self):
+ G=self.K3
+ assert_equal(sorted(G.predecessors_iter(0)),[1,2])
+ assert_raises((KeyError,networkx.NetworkXError), G.predecessors_iter,-1)
+
+
+ def test_degree(self):
+ G=self.K3
+ assert_equal(list(G.degree().values()),[4,4,4])
+ assert_equal(G.degree(),{0:4,1:4,2:4})
+ assert_equal(G.degree(0),4)
+ assert_equal(G.degree([0]),{0:4})
+ assert_raises((KeyError,networkx.NetworkXError), G.degree,-1)
+
+ def test_degree_iter(self):
+ G=self.K3
+ assert_equal(list(G.degree_iter()),[(0,4),(1,4),(2,4)])
+ assert_equal(dict(G.degree_iter()),{0:4,1:4,2:4})
+ assert_equal(list(G.degree_iter(0)),[(0,4)])
+ G.add_edge(0,1,weight=0.3,other=1.2)
+ assert_equal(list(G.degree_iter(weight='weight')),[(0,4.3),(1,4.3),(2,4)])
+ assert_equal(list(G.degree_iter(weight='other')),[(0,5.2),(1,5.2),(2,4)])
+
+
+ def test_in_degree(self):
+ G=self.K3
+ assert_equal(list(G.in_degree().values()),[2,2,2])
+ assert_equal(G.in_degree(),{0:2,1:2,2:2})
+ assert_equal(G.in_degree(0),2)
+ assert_equal(G.in_degree([0]),{0:2})
+ assert_raises((KeyError,networkx.NetworkXError), G.in_degree,-1)
+
+ def test_in_degree_iter(self):
+ G=self.K3
+ assert_equal(list(G.in_degree_iter()),[(0,2),(1,2),(2,2)])
+ assert_equal(dict(G.in_degree_iter()),{0:2,1:2,2:2})
+ assert_equal(list(G.in_degree_iter(0)),[(0,2)])
+ assert_equal(list(G.in_degree_iter(0,weight='weight')),[(0,2)])
+
+ def test_out_degree(self):
+ G=self.K3
+ assert_equal(list(G.out_degree().values()),[2,2,2])
+ assert_equal(G.out_degree(),{0:2,1:2,2:2})
+ assert_equal(G.out_degree(0),2)
+ assert_equal(G.out_degree([0]),{0:2})
+ assert_raises((KeyError,networkx.NetworkXError), G.out_degree,-1)
+
+ def test_out_degree_iter(self):
+ G=self.K3
+ assert_equal(list(G.out_degree_iter()),[(0,2),(1,2),(2,2)])
+ assert_equal(dict(G.out_degree_iter()),{0:2,1:2,2:2})
+ assert_equal(list(G.out_degree_iter(0)),[(0,2)])
+ assert_equal(list(G.out_degree_iter(0,weight='weight')),[(0,2)])
+
+
+ def test_size(self):
+ G=self.K3
+ assert_equal(G.size(),6)
+ assert_equal(G.number_of_edges(),6)
+ G.add_edge(0,1,weight=0.3,other=1.2)
+ assert_equal(G.size(weight='weight'),6.3)
+ assert_equal(G.size(weight='other'),7.2)
+
+ def test_to_undirected_reciprocal(self):
+ G=self.Graph()
+ G.add_edge(1,2)
+ assert_true(G.to_undirected().has_edge(1,2))
+ assert_false(G.to_undirected(reciprocal=True).has_edge(1,2))
+ G.add_edge(2,1)
+ assert_true(G.to_undirected(reciprocal=True).has_edge(1,2))
+
+ def test_reverse_copy(self):
+ G=networkx.MultiDiGraph([(0,1),(0,1)])
+ R=G.reverse()
+ assert_equal(sorted(R.edges()),[(1,0),(1,0)])
+ R.remove_edge(1,0)
+ assert_equal(sorted(R.edges()),[(1,0)])
+ assert_equal(sorted(G.edges()),[(0,1),(0,1)])
+
+ def test_reverse_nocopy(self):
+ G=networkx.MultiDiGraph([(0,1),(0,1)])
+ R=G.reverse(copy=False)
+ assert_equal(sorted(R.edges()),[(1,0),(1,0)])
+ R.remove_edge(1,0)
+ assert_equal(sorted(R.edges()),[(1,0)])
+ assert_equal(sorted(G.edges()),[(1,0)])
+
+
+class TestMultiDiGraph(BaseMultiDiGraphTester,TestMultiGraph):
+ def setUp(self):
+ self.Graph=networkx.MultiDiGraph
+ # build K3
+ self.k3edges=[(0, 1), (0, 2), (1, 2)]
+ self.k3nodes=[0, 1, 2]
+ self.K3=self.Graph()
+ self.K3.adj={0:{},1:{},2:{}}
+ self.K3.succ=self.K3.adj
+ self.K3.pred={0:{},1:{},2:{}}
+ for u in self.k3nodes:
+ for v in self.k3nodes:
+ if u==v: continue
+ d={0:{}}
+ self.K3.succ[u][v]=d
+ self.K3.pred[v][u]=d
+ self.K3.adj=self.K3.succ
+ self.K3.edge=self.K3.adj
+ self.K3.node={}
+ self.K3.node[0]={}
+ self.K3.node[1]={}
+ self.K3.node[2]={}
+
+
+ def test_add_edge(self):
+ G=self.Graph()
+ G.add_edge(0,1)
+ assert_equal(G.adj,{0: {1: {0:{}}}, 1: {}})
+ assert_equal(G.succ,{0: {1: {0:{}}}, 1: {}})
+ assert_equal(G.pred,{0: {}, 1: {0:{0:{}}}})
+ G=self.Graph()
+ G.add_edge(*(0,1))
+ assert_equal(G.adj,{0: {1: {0:{}}}, 1: {}})
+ assert_equal(G.succ,{0: {1: {0:{}}}, 1: {}})
+ assert_equal(G.pred,{0: {}, 1: {0:{0:{}}}})
+
+ def test_add_edges_from(self):
+ G=self.Graph()
+ G.add_edges_from([(0,1),(0,1,{'weight':3})])
+ assert_equal(G.adj,{0: {1: {0:{},1:{'weight':3}}}, 1: {}})
+ assert_equal(G.succ,{0: {1: {0:{},1:{'weight':3}}}, 1: {}})
+ assert_equal(G.pred,{0: {}, 1: {0:{0:{},1:{'weight':3}}}})
+
+ G.add_edges_from([(0,1),(0,1,{'weight':3})],weight=2)
+ assert_equal(G.succ,{0: {1: {0:{},
+ 1:{'weight':3},
+ 2:{'weight':2},
+ 3:{'weight':3}}},
+ 1: {}})
+ assert_equal(G.pred,{0: {}, 1: {0:{0:{},1:{'weight':3},
+ 2:{'weight':2},
+ 3:{'weight':3}}}})
+
+ assert_raises(networkx.NetworkXError, G.add_edges_from,[(0,)]) # too few in tuple
+ assert_raises(networkx.NetworkXError, G.add_edges_from,[(0,1,2,3,4)]) # too many in tuple
+ assert_raises(TypeError, G.add_edges_from,[0]) # not a tuple
+
+ def test_remove_edge(self):
+ G=self.K3
+ G.remove_edge(0,1)
+ assert_equal(G.succ,{0:{2:{0:{}}},
+ 1:{0:{0:{}},2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ assert_equal(G.pred,{0:{1:{0:{}}, 2:{0:{}}},
+ 1:{2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ assert_raises((KeyError,networkx.NetworkXError), G.remove_edge,-1,0)
+ assert_raises((KeyError,networkx.NetworkXError), G.remove_edge,0,2,
+ key=1)
+
+
+ def test_remove_multiedge(self):
+ G=self.K3
+ G.add_edge(0,1,key='parallel edge')
+ G.remove_edge(0,1,key='parallel edge')
+ assert_equal(G.adj,{0: {1: {0:{}}, 2: {0:{}}},
+ 1: {0: {0:{}}, 2: {0:{}}},
+ 2: {0: {0:{}}, 1: {0:{}}}})
+
+ assert_equal(G.succ,{0: {1: {0:{}}, 2: {0:{}}},
+ 1: {0: {0:{}}, 2: {0:{}}},
+ 2: {0: {0:{}}, 1: {0:{}}}})
+
+ assert_equal(G.pred,{0:{1: {0:{}},2:{0:{}}},
+ 1:{0:{0:{}},2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ G.remove_edge(0,1)
+ assert_equal(G.succ,{0:{2:{0:{}}},
+ 1:{0:{0:{}},2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ assert_equal(G.pred,{0:{1:{0:{}}, 2:{0:{}}},
+ 1:{2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ assert_raises((KeyError,networkx.NetworkXError), G.remove_edge,-1,0)
+
+ def test_remove_edges_from(self):
+ G=self.K3
+ G.remove_edges_from([(0,1)])
+ assert_equal(G.succ,{0:{2:{0:{}}},
+ 1:{0:{0:{}},2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ assert_equal(G.pred,{0:{1:{0:{}}, 2:{0:{}}},
+ 1:{2:{0:{}}},
+ 2:{0:{0:{}},1:{0:{}}}})
+ G.remove_edges_from([(0,0)]) # silent fail