def test_iter():
datadir = join(dirname(pcd.__file__), 'data')
fname = join(datadir, 'test_cmty.txt')
fname2 = join(datadir, 'test_cmty2.txt')
# Test iterator
cmtys = cmty.CommunityListIterator(fname)
#print cmtys.__dict__
assert cmtys.label == 'test-communities'
assert len(tuple(cmtys.iteritems())) == 2 # Do a full iteration
assert len(cmtys) == 2
cmty._test_interface(cmtys)
# Test conversion to full communities.
cmtys_full = cmtys.to_full()
assert cmtys_full.label == 'test-communities'
assert cmtys_full.cmtynames() == set(('one', 'two'))
#from fitz import interactnow
assert len(cmtys_full) == 2
cmty._test_interface(cmtys_full)
# Test label loading from the .names file
cmtys = cmty.CommunityFile(fname2)
assert cmtys.label == 'test-communities2'
def test_full():
# Test full object
cmtys = cmty.Communities(cmtynodes)
assert cmtys.is_cover() == True
assert cmtys.is_non_overlapping() == False
assert cmtys.is_partition() == False
cmtys.load_networkx(networkx.complete_graph(10))
# Test object with missing some nodes (non-consecutive). This is
# mainly a test of the cmtyintmap and nodeintmap.
cn2 = {
0:set((0, 1, 2, 3)),
1:set((3, 4, 5, )),
5:set((7, 8, )), #missing 6 and 9
}
cm2 = cmty.Communities(cn2)
cmty._test_interface(cm2)
def test_filter():
# Test some filters
cmtys = cmty.Communities(cmtynodes)
def filter(c, ns):
yield str(c)+'.1', ns
yield str(c)+'.2', ns
cmtys_filter = cmty.CommunityFilter(cmtys, filter)
cmtys_filter_full = cmtys_filter.to_full()
cmtys_filter_full.q == cmtys_filter.q
cmty._test_interface(cmtys_filter)
assert cmtys_filter_full.nodes_spanned() == set(range(10))
assert cmtys_filter_full.cmtysizes_sum() == 22
assert cmtys_filter_full.overlap() == 2.2
cmtys = cmty.Communities(cmtynodes)
def filter(c, ns):
if len(ns) == 4:
yield c, ns
cmtys_filter = cmty.CommunityFilter(cmtys, filter)
assert len(cmtys_filter) == 2
def test_union():
# Test unions
cn1 = dict(A=set((0, 1, 3)), B=set((1, 2, 3, 4)))
cn2 = dict(A=set((0, 1, 3)), B=set((5, 6, 7, 8, 9)))
c1 = cmty.Communities(cn1)
c2 = cmty.Communities(cn2)
cU = cmty.CommunityUnion((c1, c2))
assert len(cU) == 3
assert cU.nodes == set(range(10))
assert set(frozenset(x) for x in cU.itervalues()) \
== set((frozenset((0,1,3)),frozenset((1,2,3,4)),frozenset((5,6,7,8,9))))
#print list(cU.iteritems())
assert len(set(cU.iterkeys())) == 3
assert len(list(cU.itervalues())) == 3
cmty._test_interface(cU)
# Same test, but without dup_ok.
cU = cmty.CommunityUnion((cn1, cn2), dup_ok=True)
assert len(cU) == 4
assert len(set(cU.iterkeys())) == 4
assert cU.nodes == set(range(10))
#print list(cU.iteritems())
cmty._test_interface(cU)
def test_union():
# Test unions
cn1 = dict(A=set((0, 1, 3)), B=set((1, 2, 3, 4)))
cn2 = dict(A=set((0, 1, 3)), B=set((5, 6, 7, 8, 9)))
c1 = cmty.Communities(cn1)
c2 = cmty.Communities(cn2)
cU = cmty.CommunityUnion((c1, c2))
assert len(cU) == 3
assert cU.nodes == set(range(10))
assert set(frozenset(x) for x in cU.itervalues()) \
== set((frozenset((0,1,3)),frozenset((1,2,3,4)),frozenset((5,6,7,8,9))))
#print list(cU.iteritems())
assert len(set(cU.iterkeys())) == 3
assert len(list(cU.itervalues())) == 3
cmty._test_interface(cU)
# Same test, but without dup_ok.
cU = cmty.CommunityUnion((cn1, cn2), dup_ok = True)
assert len(cU) == 4
assert len(set(cU.iterkeys())) == 4
assert cU.nodes == set(range(10))
#print list(cU.iteritems())
cmty._test_interface(cU)
def test_filter():
# Test some filters
cmtys = cmty.Communities(cmtynodes)
def filter(c, ns):
yield str(c) + '.1', ns
yield str(c) + '.2', ns
cmtys_filter = cmty.CommunityFilter(cmtys, filter)
cmtys_filter_full = cmtys_filter.to_full()
cmtys_filter_full.q == cmtys_filter.q
cmty._test_interface(cmtys_filter)
assert cmtys_filter_full.nodes_spanned() == set(range(10))
assert cmtys_filter_full.cmtysizes_sum() == 22
assert cmtys_filter_full.overlap() == 2.2
cmtys = cmty.Communities(cmtynodes)
def filter(c, ns):
if len(ns) == 4:
yield c, ns
cmtys_filter = cmty.CommunityFilter(cmtys, filter)
assert len(cmtys_filter) == 2
def test_iter():
datadir = join(dirname(pcd.__file__), 'data')
fname = join(datadir, 'test_cmty.txt')
fname2 = join(datadir, 'test_cmty2.txt')
# Test iterator
cmtys = cmty.CommunityListIterator(fname)
#print cmtys.__dict__
assert cmtys.label == 'test-communities'
assert len(tuple(cmtys.iteritems())) == 2 # Do a full iteration
assert len(cmtys) == 2
cmty._test_interface(cmtys)
# Test conversion to full communities.
cmtys_full = cmtys.to_full()
assert cmtys_full.label == 'test-communities'
assert cmtys_full.cmtynames() == set(('one', 'two'))
#from fitz import interactnow
assert len(cmtys_full) == 2
cmty._test_interface(cmtys_full)
# Test label loading from the .names file
cmtys = cmty.CommunityFile(fname2)
assert cmtys.label == 'test-communities2'
def test_full():
# Test full object
cmtys = cmty.Communities(cmtynodes)
assert cmtys.is_cover() == True
assert cmtys.is_non_overlapping() == False
assert cmtys.is_partition() == False
cmtys.load_networkx(networkx.complete_graph(10))
# Test object with missing some nodes (non-consecutive). This is
# mainly a test of the cmtyintmap and nodeintmap.
cn2 = {
0: set((0, 1, 2, 3)),
1: set((
3,
4,
5,
)),
5: set((
7,
8,
)), #missing 6 and 9
}
cm2 = cmty.Communities(cn2)
cmty._test_interface(cm2)
