def main(): if len(sys.argv) != 2: print('File name must be provided') sys.exit(-1) filename = sys.argv[1] filenameWOExt = filename.split('.')[0].split('/')[-1] graph = snap.LoadEdgeList(snap.PNEANet, filename, 0, 1, '\t') snap.PrintInfo(graph, "New York", filenameWOExt + '_info.txt', False) Rnd = snap.TRnd(123124) erdosRenyi = snap.GenRndGnm(snap.PNEANet, graph.GetNodes(), graph.GetEdges(), True, Rnd) snap.PrintInfo(erdosRenyi, "Erdos-Renyi", 'erdos_renyi_info.txt', False) grid = snap.GenGrid(snap.PNEANet, 220, 250, False) snap.PrintInfo(grid, "Grid", 'grid_info.txt', False) printGenericInformation(graph, 'New York street network') printGenericInformation(erdosRenyi, 'Erdos-Renyi random graph') printGenericInformation(grid, 'Grid random graph') # Plot everything in the plots directory os.chdir(os.path.join(os.path.abspath(sys.path[0]), 'plots')) saveDegreeDistribution(graph, 'deg_dist_ny.tab') saveDegreeDistribution(erdosRenyi, 'deg_dist_er.tab') saveDegreeDistribution(grid, 'deg_dist_gr.tab') testRobustnessAll([graph, erdosRenyi, grid]) call(['gnuplot', 'deg_dist.plt']) call(['gnuplot', 'robustness_rand.plt']) call(['gnuplot', 'robustness_max.plt'])
def test_automorphismCount(): H = snap.GenGrid(snap.PUNGraph, 3, 2) assert (automorphismCount(H) == 4) H = snap.GenStar(snap.PUNGraph, 5) assert (automorphismCount(H) == 24)
#!/usr/bin/env python3 import snap import sys if len(sys.argv) < 3: print("Usage mode: ./{} height width".format(sys.argv[0])) sys.exit(1) h = int(sys.argv[1]) w = int(sys.argv[2]) # my_seed = 42 G = snap.GenGrid(snap.TNGraph, h, w) file_name = "grid_h{}_w{}.txt".format(h, w) f = open(file_name, "w") for e in G.Edges(): (u, v) = (e.GetSrcNId(), e.GetDstNId()) f.write("{}\t{}\n".format(u, v)) f.close()
import snap G = snap.GenGrid(snap.PUNGraph, 5, 3) snap.DrawGViz(G, snap.gvlDot, "grid5x3.png", "Grid 5x3")
def test_graphviz(): G = snap.GenGrid(snap.PUNGraph, 5, 3) G.DrawGViz(snap.gvlDot, "grid5x3.png", "Grid 5x3")