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)