tests in. run

Author: RobRossmiller-TG

tests/data/unweighted edges/empty_graph_edges.csv renamed to .github/workflows/tests.yml

@@ -0,0 +1,38 @@
+[
+  {
+    "@@top_scores_heap": [
+      {
+        "Vertex_ID": "A",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "B",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "C",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "D",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "E",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "F",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "G",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "H",
+        "score": 1.2857142857142856
+      }
+    ]
+  }
+]

tests/data/baseline/graph_algorithms_baselines/centrality/degree_centrality/CompleteUnweighted.json

@@ -0,0 +1,38 @@
+[
+  {
+    "@@top_scores_heap": [
+      {
+        "Vertex_ID": "A",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "B",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "C",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "D",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "E",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "F",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "G",
+        "score": 1.2857142857142856
+      },
+      {
+        "Vertex_ID": "H",
+        "score": 1.2857142857142856
+      }
+    ]
+  }
+]

tests/data/baseline/graph_algorithms_baselines/centrality/degree_centrality/CompleteWeighted.json

@@ -0,0 +1,59 @@
+import csv
+import json
+
+import networkx as nx
+import numpy as np
+
+baseline_path_root = "baseline/graph_algorithms_baselines"
+
+
+def run_degree_baseline(g: nx.Graph):
+    res = nx.centrality.degree_centrality(g)
+    nx.centrality.degree_centrality
+
+    out = []
+    for k, v in res.items():
+        out.append({"Vertex_ID": k, "score": v})
+
+    out = [{"@@top_scores_heap": out}]
+    return out
+
+
+def create_graph(path, edges, weights):
+    g = nx.Graph()
+    # include edge weights if they exist
+    if weights is not None:
+        g.add_weighted_edges_from(edges)
+    else:
+        g.add_edges_from(edges)
+    return g
+
+
+def create_degree_baseline():
+    # input, output
+    paths = [
+        (
+            "unweighted_edges/complete_edges.csv",
+            f"{baseline_path_root}/centrality/degree_centrality/CompleteUnweighted.json",
+            False,
+        ),
+        (
+            "weighted_edges/complete_edges.csv",
+            f"{baseline_path_root}/centrality/degree_centrality/CompleteWeighted.json",
+            True,
+        ),
+    ]
+
+    for p, o_path, w in paths:
+        with open(p) as f:
+            edges = np.array(list(csv.reader(f)))
+
+        g = create_graph(p, edges, w)
+
+        res = run_degree_baseline(g)
+        with open(o_path, "w") as f:
+            json.dump(res, f, indent=2)
+
+
+if __name__ == "__main__":
+    create_degree_baseline()

tests/data/complete.png

@@ -21,13 +21,15 @@ class TestCentrality:
         "Ring_Weighted",
         "Hub_Spoke_Weighted",
         "Tree_Weighted",
+        "CompleteWeighted",
     ]
     # weighted directed graphs
     graph_types4 = [
         "Line_Directed_Weighted",
         "Ring_Directed_Weighted",
         "Hub_Spoke_Directed_Weighted",
         "Tree_Directed_Weighted",
+        "CompleteUnweighted",
     ]
 
     @pytest.mark.parametrize("test_name", graph_types1)