from pymote.algorithms.buildTreeKVM import BuildTree
from pymote.networkgenerator import NetworkGenerator
from pymote.simulation import Simulation
from pymote.network import Network
import networkx as nx

#G = nx.gnm_random_graph(5,5)
#net_gen = NetworkGenerator(5)
#net = net_gen.generate_random_network()
net = Network()
net.add_weighted_edges_from([(1,2,8), (3,4,6), (5,6,7)])
net.algorithms = (BuildTree,)
net.show()

sim = Simulation(net)
sim.run()

for node in net.nodes():
    print node.id, node.memory
#sim.reset()
print "\nDone script."