def test_clear_weights(self): # create new topology to avoid parameters pollution G = fnss.star_topology(12) fnss.set_weights_constant(G, 3, None) self.assertEqual(G.number_of_edges(), len(nx.get_edge_attributes(G, 'weight'))) fnss.clear_weights(G) self.assertEqual(0, len(nx.get_edge_attributes(G, 'weight')))
def tearDown(self): fnss.clear_weights(self.topo)