def setUp(self):
self.network = WirelessNetwork()
# create a settings object
self.settings = Settings()
# set the initial energy level of a node
self.settings.xii = 10.0
# store the settings
self.network.settings = self.settings
serialize = GraphSerialize()
# set up a simple network
self.network.network = serialize.read_dot_file('grid.dot')
# set file handler
self.network.file_handler = logging.FileHandler('unittest.log')
# enable file handler
self.network.logger.addHandler(self.network.file_handler)
class TestWirelessNetworkFunctions(unittest.TestCase):
def setUp(self):
self.network = WirelessNetwork()
# create a settings object
self.settings = Settings()
# set the initial energy level of a node
self.settings.xii = 10.0
# store the settings
self.network.settings = self.settings
serialize = GraphSerialize()
# set up a simple network
self.network.network = serialize.read_dot_file('grid.dot')
# set file handler
self.network.file_handler = logging.FileHandler('unittest.log')
# enable file handler
self.network.logger.addHandler(self.network.file_handler)
def test_setup(self):
# test the setup method
self.network.setup()
# every node should have a routing table
for node in self.network.network.nodes():
# check if a routing table instance was created
self.assertIsInstance(self.network.network.node[node]['routing table'], RoutingTable)
# the energy level should match the energy level defined in the settings
self.assertIs(self.network.network.node[node]['energy'], self.settings.xii)
# check if the active flag was set correctly
self.assertTrue(self.network.network.node[node]['active'])
# check if the array was initialized
self.assertIsNotNone(self.network.network.node[node]['last packets'])
def test_initialize_route_discovery(self):
src, dst = 4, 10
# set up the network
self.network.setup()
# initialize route discovery (test the method)
self.network.initialize_route_discovery(src, dst)
# print self.network.dumpRoutingTable()
def test_random_select(self):
# set up the network
self.network.setup()
# self.network.find_pair(4)
def test_random_select_hard(self):
serialize = GraphSerialize()
self.network.network = serialize.read_dot_file('advanced.dot')
# set up the network
self.network.setup()
self.network.find_pair(8)
