def test_set_compositeSensor(self):
"""Test setting compositeSensors on a node"""
net = Network()
node = net.add_node()
dist_sensor = DistSensor({'pf': scipy.stats.norm, 'scale': 10})
node.compositeSensor = (NeighborsSensor, 'AoASensor', dist_sensor)
self.assertTrue(len(node.compositeSensor.sensors) == 3)
readings = node.compositeSensor.read()
self.assertTrue('Neighbors' in readings.keys() and
'AoA' in readings.keys() and
'Dist' in readings.keys())
def _create_modify_network(self, net=None, step=1):
"""Helper method for creating new or modifying given network.
Arguments:
net (int):
network to modify, if None create from scratch
step:
if >0 new network should be more dense for <0 less dense
"""
if net is None:
net = Network(**self.kwargs)
for _n in range(self.n_count):
node = Node(commRange=self.comm_range, **self.kwargs)
net.add_node(node)
else:
print "here2-" + str(step)
if step>0:
if len(net)<self.n_max:
node = Node(**self.kwargs)
net.add_node(node)
logger.debug("Added node, number of nodes: %d (%d)"
% (len(net), int(self.n_max)))
elif not self.comm_range:
for node in net.nodes():
node.commRange += step
logger.debug("Increased commRange to %d"
% node.commRange)
else:
return None
else:
print "here3-" + str(self.n_min)
if len(net) > self.n_min and len(net) > 1:
net.remove_node(net.nodes()[0])
logger.debug("Removed node, nodes left: %d"
% len(net))
elif not self.comm_range:
if abs(step) >= net.nodes()[0].commRange:
step /= 4
for node in net:
node.commRange += step
logger.debug("Decreased commRange to %d"
% net.nodes()[0].commRange)
else:
return None
return net
def test_read(self):
"""Test read compositeSensor"""
net = Network()
node = net.add_node()
node.compositeSensor.read()
#show() net.adj = g.adj net.show() #Uses Kruskal’s algorithm. #If the graph edges do not have a weight attribute #a default weight of 1 will be used. mst = minimum_spanning_tree(net) net.adj = mst.adj net.show() from pymote.network import Network net = Network() node = net.add_node(pos=[200, 300]) node = net.add_node(pos=[300, 300]) node = net.add_node(pos=[100, 200]) node = net.add_node(pos=[400, 200]) node = net.add_node(pos=[200, 100]) node = net.add_node(pos=[300, 100]) a = Graph() a.add_edge(net.nodes()[0], net.nodes()[1], weight=1.1) a.add_edge(net.nodes()[0], net.nodes()[2], weight=1.7) a.add_edge(net.nodes()[0], net.nodes()[4], weight=2.6) a.add_edge(net.nodes()[1], net.nodes()[3], weight=3.1) a.add_edge(net.nodes()[2], net.nodes()[4], weight=3.8) a.add_edge(net.nodes()[3], net.nodes()[5], weight=3.7)
def setUp(self):
self.net = Network()
self.net.environment.im[22, 22] = 0
self.node1 = self.net.add_node(pos=[22.8, 21.8])
self.node2 = self.net.add_node(pos=[21.9, 22.9])
self.node3 = self.net.add_node(pos=[21.7, 21.7])
