1 files changed, 30 insertions, 0 deletions
diff --git a/lib/python2.7/site-packages/setoolsgui/networkx/algorithms/tests/test_hierarchy.py b/lib/python2.7/site-packages/setoolsgui/networkx/algorithms/tests/test_hierarchy.py
new file mode 100644
index 0000000..6f2907e
--- /dev/null
+++ b/lib/python2.7/site-packages/setoolsgui/networkx/algorithms/tests/test_hierarchy.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+from nose.tools import *
+import networkx as nx
+
+def test_hierarchy_exception():
+    G = nx.cycle_graph(5)
+    assert_raises(nx.NetworkXError,nx.flow_hierarchy,G)
+
+def test_hierarchy_cycle():
+    G = nx.cycle_graph(5,create_using=nx.DiGraph())
+    assert_equal(nx.flow_hierarchy(G),0.0)
+
+def test_hierarchy_tree():
+    G = nx.full_rary_tree(2,16,create_using=nx.DiGraph())
+    assert_equal(nx.flow_hierarchy(G),1.0)
+
+def test_hierarchy_1():
+    G = nx.DiGraph()
+    G.add_edges_from([(0,1),(1,2),(2,3),(3,1),(3,4),(0,4)])
+    assert_equal(nx.flow_hierarchy(G),0.5)
+
+def test_hierarchy_weight():
+    G = nx.DiGraph()
+    G.add_edges_from([(0,1,{'weight':.3}),
+                      (1,2,{'weight':.1}),
+                      (2,3,{'weight':.1}),
+                      (3,1,{'weight':.1}),
+                      (3,4,{'weight':.3}),
+                      (0,4,{'weight':.3})])
+    assert_equal(nx.flow_hierarchy(G,weight='weight'),.75)
+\ No newline at end of file