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
27
28
29
30
31
32
33
34
35
36
37
38
|
"""
Unit tests for matplotlib drawing functions.
"""
import os, sys
import tempfile
from nose import SkipTest
from nose.tools import assert_true
import networkx as nx
class TestPylab(object):
@classmethod
def setupClass(cls):
global pylab
try:
import pylab
except ImportError:
raise SkipTest('matplotlib not available.')
def setUp(self):
self.G=nx.barbell_graph(5,10)
def test_draw(self):
# hold(False)
N=self.G
nx.draw_spring(N)
pylab.savefig("test.png")
nx.draw_random(N)
pylab.savefig("test.ps")
nx.draw_circular(N)
pylab.savefig("test.png")
nx.draw_spectral(N)
pylab.savefig("test.png")
os.unlink('test.png')
os.unlink('test.ps')
|
