def test_loop_erased_random_walk_sampler():
"""
Testing the number of nodes, connectivity and tree structure.
"""
sampler = LoopErasedRandomWalkSampler()
graph = nx.watts_strogatz_graph(200, 10, 0)
new_graph = sampler.sample(graph)
assert new_graph.number_of_edges() + 1 == new_graph.number_of_nodes()
assert sampler.number_of_nodes == new_graph.number_of_nodes()
assert nx.is_connected(new_graph)
assert type(new_graph) == nx.classes.graph.Graph
sampler = LoopErasedRandomWalkSampler(number_of_nodes=25)
graph = nx.watts_strogatz_graph(100, 10, 0)
new_graph = sampler.sample(graph)
assert new_graph.number_of_edges() + 1 == new_graph.number_of_nodes()
assert sampler.number_of_nodes == new_graph.number_of_nodes()
assert nx.is_connected(new_graph)
assert type(new_graph) == nx.classes.graph.Graph
def test_loop_erased_random_walk_sampler_3():
sampler = LoopErasedRandomWalkSampler()
graph = nk.nxadapter.nx2nk(nx.watts_strogatz_graph(200, 10, 0))
new_graph = sampler.sample(graph)
assert new_graph.numberOfEdges()+1 == new_graph.numberOfNodes()
assert sampler.number_of_nodes == new_graph.numberOfNodes()
assert 1 == nk.components.ConnectedComponents(new_graph).run().numberOfComponents()
assert type(new_graph) == NKGraph
new_graph = sampler.sample(graph, 0)
assert new_graph.numberOfEdges()+1 == new_graph.numberOfNodes()
assert sampler.number_of_nodes == new_graph.numberOfNodes()
assert 1 == nk.components.ConnectedComponents(new_graph).run().numberOfComponents()
assert type(new_graph) == NKGraph
new_graph = sampler.sample(graph, 199)
assert new_graph.numberOfEdges()+1 == new_graph.numberOfNodes()
assert sampler.number_of_nodes == new_graph.numberOfNodes()
assert 1 == nk.components.ConnectedComponents(new_graph).run().numberOfComponents()
assert type(new_graph) == NKGraph
new_graph = sampler.sample(graph, 20)
assert new_graph.numberOfEdges()+1 == new_graph.numberOfNodes()
assert sampler.number_of_nodes == new_graph.numberOfNodes()
assert 1 == nk.components.ConnectedComponents(new_graph).run().numberOfComponents()
assert type(new_graph) == NKGraph
def test_loop_erased_random_walk_sampler_2():
sampler = LoopErasedRandomWalkSampler(number_of_nodes=25)
graph = nx.watts_strogatz_graph(100, 10, 0)
new_graph = sampler.sample(graph)
assert new_graph.number_of_edges() + 1 == new_graph.number_of_nodes()
assert sampler.number_of_nodes == new_graph.number_of_nodes()
assert nx.is_connected(new_graph)
assert type(new_graph) == NXGraph
"""Loop erased random walk sampler example""" import networkx as nx from littleballoffur.exploration_sampling import LoopErasedRandomWalkSampler graph = nx.watts_strogatz_graph(1000, 10, 0) sampler = LoopErasedRandomWalkSampler() new_graph = sampler.sample(graph)