def test_draw_mlayer_nonaligned_mlayer_coords(self):
nc=visuals.layouts.get_fruchterman_reingold_multilayer_layout(self.mlayer_nonaligned_aligntest)
fig=visuals.draw(self.mlayer_nonaligned_aligntest,nodeCoords=nc)
fig.savefig(os.path.join(self.figdirpath,"mlayer_nonaligned_mlayer_coords.png"))
nc2=visuals.layouts.get_fruchterman_reingold_multilayer_layout(self.mlayer_nonaligned_aligntest2)
fig2=visuals.draw(self.mlayer_nonaligned_aligntest2,nodeCoords=nc2)
fig2.savefig(os.path.join(self.figdirpath,"mlayer_nonaligned_mlayer_coords2.png"))
nc3=visuals.layouts.get_fruchterman_reingold_multilayer_layout(self.mlayer_nonaligned_aligntest2,alignedNodes=False)
fig3=visuals.draw(self.mlayer_nonaligned_aligntest2,nodelayerCoords=nc3)
fig3.savefig(os.path.join(self.figdirpath,"mlayer_nonaligned_mlayer_coords3.png"))
def test_multiaxis(self):
from matplotlib import pyplot as plt
fig = plt.figure()
ax1=fig.add_subplot(121, projection='3d')
ax2=fig.add_subplot(122, projection='3d')
nc=visuals.layouts.get_fruchterman_reingold_multilayer_layout(self.mlayer_nonaligned_aligntest)
visuals.draw(self.mlayer_nonaligned_aligntest,nodeCoords=nc,ax=ax1)
nc2=visuals.layouts.get_fruchterman_reingold_multilayer_layout(self.mlayer_nonaligned_aligntest2)
visuals.draw(self.mlayer_nonaligned_aligntest2,nodeCoords=nc2,ax=ax2)
fig.savefig(os.path.join(self.figdirpath,"multiaxis_mlayer.png"))
def test_mplex_networkx_layouts(self):
from pymnet import nx
g=nx.karate_club_graph()
mplex=net.MultiplexNetwork()
mplex.add_layer("karate-1")
mplex.add_layer("karate-2")
mplex.A['karate-1']=g
mplex.A['karate-2']=g
fig=visuals.draw(mplex,layout="spring")
fig.savefig(os.path.join(self.figdirpath,"mplex_networkx_spring.png"))
def test_draw_mplex_simple_layer_labels(self):
fig = visuals.draw(self.mplex_simple,
layerLabelColorDict={
1: "blue",
2: "green"
},
layerLabelSizeRule={
"rule": "name",
"scaleby": 10
},
layerLabelAlphaDict={3: 0.5},
layerLabelStyleDict={2: "italic"})
fig.savefig(
os.path.join(self.figdirpath,
"mlayer_example_1d_layer_labels.png"))
def test_draw_assigners_advanced1(self):
fig = visuals.draw(self.mplex_simple,
edgeWidthRule={
"rule": "edgeweight",
"scaleby": "layer",
1: 1.0,
2: 0.5,
3: 2.0,
"interlayer": 3
},
nodeColorRule={
"rule": "layer",
"mapping": True,
1: "red",
2: "blue",
3: "green"
})
fig.savefig(
os.path.join(self.figdirpath,
"mlayer_example_1d_assigners_advanced1.png"))
def test_mplex_fr_layout(self):
from pymnet import models
mplex=models.er(10,2*[0.2])
fig=visuals.draw(mplex,layout="fr")
fig.savefig(os.path.join(self.figdirpath,"mplex_er100_fr.png"))
def test_draw_mlayer_example_1d_defaults(self):
fig=visuals.draw(self.mlayer_example_1d)
fig.savefig(os.path.join(self.figdirpath,"mlayer_example_1d_defaults.png"))
def test_draw_mplex_nonaligned_simple_defaults(self):
fig=visuals.draw(self.mplex_nonaligned_simple)
fig.savefig(os.path.join(self.figdirpath,"mplex_nonaligned_simple_defaults.png"))