Coverage for src/abm_colony_collection/calculate_degree_measures.py: 100%

1import networkx as nx 

2import pandas as pd 

3 

4 

5def calculate_degree_measures(network: nx.Graph) -> pd.DataFrame: 

6 """ 

7 Calculate degree measures for each node in network. 

8 

9 Measures include: 

10 

11 - Degree = number of edges adjacent to the node 

12 

13 Parameters 

14 ---------- 

15 network 

16 The network object. 

17 

18 Returns 

19 ------- 

20 : 

21 Degree measures for each node in the network. 

22 """ 

23 

24 # Extract degree for each node in network. 

25 measures = [ 

26 { 

27 "ID": node, 

28 "DEGREE": degree, 

29 } 

30 for node, degree in network.degree() 

31 ] 

32 

33 return pd.DataFrame(measures)