| import networkx as nx |
| from nose.tools import * |
| |
| |
| def test_richclub(): |
| G = nx.Graph([(0,1),(0,2),(1,2),(1,3),(1,4),(4,5)]) |
| rc = nx.richclub.rich_club_coefficient(G,normalized=False) |
| assert_equal(rc,{0: 12.0/30,1:8.0/12}) |
| |
| # test single value |
| rc0 = nx.richclub.rich_club_coefficient(G,normalized=False)[0] |
| assert_equal(rc0,12.0/30.0) |
| |
| def test_richclub_normalized(): |
| G = nx.Graph([(0,1),(0,2),(1,2),(1,3),(1,4),(4,5)]) |
| rcNorm = nx.richclub.rich_club_coefficient(G,Q=2) |
| assert_equal(rcNorm,{0:1.0,1:1.0}) |
| |
| |
| def test_richclub2(): |
| T = nx.balanced_tree(2,10) |
| rc = nx.richclub.rich_club_coefficient(T,normalized=False) |
| assert_equal(rc,{0:4092/(2047*2046.0), |
| 1:(2044.0/(1023*1022)), |
| 2:(2040.0/(1022*1021))}) |
| |
| #def test_richclub2_normalized(): |
| # T = nx.balanced_tree(2,10) |
| # rcNorm = nx.richclub.rich_club_coefficient(T,Q=2) |
| # assert_true(rcNorm[0] ==1.0 and rcNorm[1] < 0.9 and rcNorm[2] < 0.9) |