def test_json_graph(self):
export = GraphSerialize()
export.write_json_graph("unittest.json", self.network)
read_network = export.read_json_graph("unittest.json")
# check if the written file has the number of nodes (and identifier)
self.assertEqual(self.network.nodes(), read_network.nodes())
# check if the first node has the same neighbors
self.assertEqual(self.network.neighbors(1), read_network.neighbors(1))
# pick a random node
random_node = int(random.uniform(2,100))
# check if a randomly picked node has the same neighbors
self.assertEqual(self.network.nodes(random_node), read_network.nodes(random_node))
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_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)
