1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
#!/usr/bin/env python
from nose.tools import *
import networkx
class TestBipartite:
def test_project(self):
G=networkx.path_graph(4)
P=networkx.project(G,[1,3])
assert_equal(sorted(P.nodes()),[1,3])
assert_equal(sorted(P.edges()),[(1,3)])
def test_is_bipartite(self):
G=networkx.path_graph(4)
assert_equal(networkx.is_bipartite(G),True)
def test_bipartite_color(self):
G=networkx.path_graph(4)
c=networkx.bipartite_color(G)
assert_equal(c,{0: 1, 1: 0, 2: 1, 3: 0})
def test_bipartite_sets(self):
G=networkx.path_graph(4)
X,Y=networkx.bipartite_sets(G)
assert_equal(X,set([0,2]))
assert_equal(Y,set([1,3]))
|
