def test_induction_samplers():
"""
Testing the density of induced subgraphs
"""
induced_sampler = RandomEdgeSamplerWithInduction()
partially_induced_sampler = RandomEdgeSamplerWithPartialInduction()
graph = nx.watts_strogatz_graph(200, 10, 0)
induced_graph = induced_sampler.sample(graph)
partially_induced_graph = partially_induced_sampler.sample(graph)
assert nx.density(partially_induced_graph) <= nx.density(induced_graph)
def test_induction_samplers():
"""
Testing the density of induced subgraphs
"""
induced_sampler = RandomEdgeSamplerWithInduction()
partially_induced_sampler = RandomEdgeSamplerWithPartialInduction()
graph = nx.watts_strogatz_graph(200, 10, 0)
induced_graph = induced_sampler.sample(graph)
partially_induced_graph = partially_induced_sampler.sample(graph)
assert nx.density(partially_induced_graph) <= nx.density(induced_graph)
assert type(induced_graph) == NXGraph
assert type(partially_induced_graph) == NXGraph
induced_sampler = RandomEdgeSamplerWithInduction()
partially_induced_sampler = RandomEdgeSamplerWithPartialInduction()
graph = nx.watts_strogatz_graph(100, 10, 0)
induced_graph = induced_sampler.sample(graph)
partially_induced_graph = partially_induced_sampler.sample(graph)
assert nx.density(partially_induced_graph) <= nx.density(induced_graph)
assert type(induced_graph) == NXGraph
assert type(partially_induced_graph) == NXGraph
induced_sampler = RandomEdgeSamplerWithInduction()
partially_induced_sampler = RandomEdgeSamplerWithPartialInduction()
graph = nk.generators.WattsStrogatzGenerator(200, 10, 0.0).generate()
induced_graph = induced_sampler.sample(graph)
partially_induced_graph = partially_induced_sampler.sample(graph)
assert type(induced_graph) == NKGraph
assert type(partially_induced_graph) == NKGraph
induced_sampler = RandomEdgeSamplerWithInduction()
partially_induced_sampler = RandomEdgeSamplerWithPartialInduction()
graph = nk.generators.WattsStrogatzGenerator(100, 10, 0.0).generate()
induced_graph = induced_sampler.sample(graph)
partially_induced_graph = partially_induced_sampler.sample(graph)
assert type(induced_graph) == NKGraph
assert type(partially_induced_graph) == NKGraph
new_graph = sampler.sample(graph) #------------------------------------------- # Common Neighbor Aware Random Walk Sampler #------------------------------------------- sampler = CommonNeighborAwareRandomWalkSampler() new_graph = sampler.sample(graph) #-------------------------------------------- # Random Edge Sampler With Partial Induction #-------------------------------------------- sampler = RandomEdgeSamplerWithPartialInduction() new_graph = sampler.sample(graph) #------------------------ # Shortest Path Sampler #------------------------ sampler = ShortestPathSampler() new_graph = sampler.sample(graph) #------------------------------------------- # Circulated Neighbors Random Walk Sampler #-------------------------------------------
"""Random edge sampler with partial induction example.""" import networkx as nx from littleballoffur.edge_sampling import RandomEdgeSamplerWithPartialInduction graph = nx.watts_strogatz_graph(1000, 10, 0) sampler = RandomEdgeSamplerWithPartialInduction() new_graph = sampler.sample(graph)