def test_symmnet(self):
fig = pylab.figure()
ax = fig.add_subplot(111)
visuals.visualizeNet(self.symmnet,
axes=ax,
frame=True,
defaultNodeColor='white',
defaultNodeEdgeColor='blue',
defaultEdgeColor='red',
defaultNodeSize=20,
defaultEdgeWidth=2,
defaultNodeEdgeWidth=2.0,
labelAllNodes=True,
defaultLabelPosition='in')
fig.savefig(self.folder + "test_symmnet.eps")
def test_dirnet(self):
fig = pylab.figure(figsize=(6, 6 / np.sqrt(2)))
ax = fig.add_axes([0.2, 0.2, 0.2, 0.2])
coords = {
0: (0, 0),
1: (2, 0),
2: (-2 / np.sqrt(2), 2 / np.sqrt(2)),
3: (-2 / np.sqrt(2), -2 / np.sqrt(2))
}
visuals.visualizeNet(self.dirnet,
axes=ax,
frame=True,
coords=coords,
margin=0.05,
defaultNodeColor='white',
defaultNodeEdgeColor='blue',
defaultEdgeColor='red',
defaultNodeSize=6,
defaultEdgeWidth=0.1,
defaultNodeEdgeWidth=0.4,
labelAllNodes=True,
defaultLabelPosition='in',
labelAllEdges=True)
fig.savefig(self.folder + "test_dirnet.eps")