added remove_selfloop_edges

network_diffusion/mln/functions.py

@@ -544,3 +544,12 @@ def draw_mln(net: MultilayerNetwork, dpi: int = 300) -> None:
         nx.draw(layer_graph, ax=axs[idx], pos=pos)
         nx.drawing.draw_networkx_labels(layer_graph, ax=axs[idx], pos=pos)
     plt.show()
+
+
+def remove_selfloop_edges(net: MultilayerNetwork) -> MultilayerNetwork:
+    """Remove selfloop edges from the network."""
+    for l_name in net.layers:
+        selfloop_edges = list(nx.selfloop_edges(net.layers[l_name]))
+        net.layers[l_name].remove_edges_from(selfloop_edges)
+
+    return net

network_diffusion/tests/mln/test_functions.py

@@ -10,6 +10,7 @@
     get_toy_network_piotr,
     multiplexing_coefficient,
     neighbourhood_size,
+    remove_selfloop_edges,
 )
 
 DEGREE_NET_1 = {
@@ -125,6 +126,15 @@ def test_neighbourhood_size(self):
             raw_nss = neighbourhood_size(network)
             filtered_nss = {a.actor_id: d for a, d in raw_nss.items()}
             self.assertEqual(filtered_nss, exp_neighbourhood_size)
+
+    def test_remove_selfloop_edges(self):
+        self.network_1["business"].add_edge('Medici', 'Medici')
+        self.network_1["marriage"].add_edge('Ridolfi', 'Ridolfi')
+
+        self.network_1 = remove_selfloop_edges(self.network_1)
+
+        for l_name in self.network_1.layers:
+            self.assertEqual(list(nx.selfloop_edges(self.network_1[l_name])), [])
 
 
 if __name__ == "__main__":