test.py

import Graph_Sampling
import networkx as nx

# read facebook network edge list "fb.txt" and return a graph g.
g = nx.read_edgelist("fb.txt", create_using=nx.Graph(), nodetype=int)

# make an object and call function SRW
object1 = Graph_Sampling.SRW_RWF_ISRW()
sample1 = object1.random_walk_sampling_simple(
    g, 100
)  # graph, number of nodes to sample
print("Simple Random Walk Sampling:")
print("Number of nodes sampled=", len(sample1.nodes()))
print("Number of edges sampled=", len(sample1.edges()))

# make an object and call function RWF
object2 = Graph_Sampling.SRW_RWF_ISRW()
sample2 = object2.random_walk_sampling_with_fly_back(
    g, 110, 0.2
)  # graph, number of nodes to sample, fly-back probability
print("Random Walk Sampling with flyback:")
print("Number of nodes sampled=", len(sample2.nodes()))
print("Number of edges sampled=", len(sample2.edges()))

# make an object and call function ISRW
object3 = Graph_Sampling.SRW_RWF_ISRW()
sample3 = object3.random_walk_induced_graph_sampling(
    g, 120
)  # graph, number of nodes to sample
print("Induced Subgraph Random Walk Sampling:")
print("Number of nodes sampled=", len(sample3.nodes()))
print("Number of edges sampled=", len(sample3.edges()))

# make an object and call function SB
object3 = Graph_Sampling.Snowball()
sample3 = object3.snowball(g, 28000, 25)  # graph, number of nodes to sample , k set
print("Snowball Sampling:")
print("Number of nodes sampled=", len(sample3.nodes()))
print("Number of edges sampled=", len(sample3.edges()))

# make an object and call function FF
object4 = Graph_Sampling.ForestFire()
sample4 = object4.forestfire(g, 28000)  # graph, number of nodes to sample
print("Forest Fire Sampling")
print("Number of nodes sampled=", len(sample4.nodes()))
print("Number of edges sampled=", len(sample4.edges()))

# make an object and call function MHRW
object5 = Graph_Sampling.MHRW()
sample5 = object5.mhrw(
    g, 28000, 30
)  # graph, number of nodes to sample, initial seed node
print("Metropolis Hasting Random Walk Sampling:")
print("Number of nodes sampled=", len(sample5.nodes()))
print("Number of edges sampled=", len(sample5.edges()))

# make an object and call function TIES
object6 = Graph_Sampling.TIES()
sample6 = object6.ties(g, 10, 0.01)  # graph, number of nodes to sample, phi
print("TIES:")
print("Number of nodes sampled=", len(sample6.nodes()))
print("Number of edges sampled=", len(sample6.edges()))