tests in. run

tigergraph · May 21, 2024 · b494116 · b494116
1 parent ce724c0
commit b494116
Show file tree

Hide file tree

Showing 27 changed files with 137 additions and 0 deletions.
diff --git a/...ta/unweighted edges/empty_graph_edges.csv → .github/workflows/tests.yml b/...ta/unweighted edges/empty_graph_edges.csv → .github/workflows/tests.yml
diff --git a/.../baseline/graph_algorithms_baselines/centrality/degree_centrality/CompleteUnweighted.json b/.../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
+      }
+    ]
+  }
+]
diff --git a/...ta/baseline/graph_algorithms_baselines/centrality/degree_centrality/CompleteWeighted.json b/...ta/baseline/graph_algorithms_baselines/centrality/degree_centrality/CompleteWeighted.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
+      }
+    ]
+  }
+]
diff --git a/tests/data/complete.png b/tests/data/complete.png
diff --git a/tests/data/create_baseline.py b/tests/data/create_baseline.py
@@ -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()
diff --git a/.../data/unweighted edges/complete_edges.csv → .../data/unweighted_edges/complete_edges.csv b/.../data/unweighted edges/complete_edges.csv → .../data/unweighted_edges/complete_edges.csv
diff --git a/...eighted edges/complete_edges_directed.csv → ...eighted_edges/complete_edges_directed.csv b/...eighted edges/complete_edges_directed.csv → ...eighted_edges/complete_edges_directed.csv
diff --git a/tests/data/unweighted edges/dag_edges.csv → tests/data/unweighted_edges/dag_edges.csv b/tests/data/unweighted edges/dag_edges.csv → tests/data/unweighted_edges/dag_edges.csv
diff --git a/...data/weighted edges/empty_graph_edges.csv → ...ta/unweighted_edges/empty_graph_edges.csv b/...data/weighted edges/empty_graph_edges.csv → ...ta/unweighted_edges/empty_graph_edges.csv
diff --git a/... edges/hubspoke_connected_spoke_edges.csv → ..._edges/hubspoke_connected_spoke_edges.csv b/... edges/hubspoke_connected_spoke_edges.csv → ..._edges/hubspoke_connected_spoke_edges.csv
diff --git a/.../data/unweighted edges/hubspoke_edges.csv → .../data/unweighted_edges/hubspoke_edges.csv b/.../data/unweighted edges/hubspoke_edges.csv → .../data/unweighted_edges/hubspoke_edges.csv
diff --git a/tests/data/unweighted edges/line_edges.csv → tests/data/unweighted_edges/line_edges.csv b/tests/data/unweighted edges/line_edges.csv → tests/data/unweighted_edges/line_edges.csv
diff --git a/...ted edges/mulithub_shared_spoke_edges.csv → ...ted_edges/mulithub_shared_spoke_edges.csv b/...ted edges/mulithub_shared_spoke_edges.csv → ...ted_edges/mulithub_shared_spoke_edges.csv
diff --git a/tests/data/unweighted edges/ring_edges.csv → tests/data/unweighted_edges/ring_edges.csv b/tests/data/unweighted edges/ring_edges.csv → tests/data/unweighted_edges/ring_edges.csv
diff --git a/tests/data/unweighted edges/tree_edges.csv → tests/data/unweighted_edges/tree_edges.csv b/tests/data/unweighted edges/tree_edges.csv → tests/data/unweighted_edges/tree_edges.csv
diff --git a/tests/data/weighted edges/complete_edges.csv → tests/data/weighted_edges/complete_edges.csv b/tests/data/weighted edges/complete_edges.csv → tests/data/weighted_edges/complete_edges.csv
diff --git a/...eighted edges/complete_edges_directed.csv → ...eighted_edges/complete_edges_directed.csv b/...eighted edges/complete_edges_directed.csv → ...eighted_edges/complete_edges_directed.csv
diff --git a/tests/data/weighted edges/dag_edges.csv → tests/data/weighted_edges/dag_edges.csv b/tests/data/weighted edges/dag_edges.csv → tests/data/weighted_edges/dag_edges.csv
diff --git a/tests/data/weighted_edges/empty_graph_edges.csv b/tests/data/weighted_edges/empty_graph_edges.csv
diff --git a/... edges/hubspoke_connected_spoke_edges.csv → ..._edges/hubspoke_connected_spoke_edges.csv b/... edges/hubspoke_connected_spoke_edges.csv → ..._edges/hubspoke_connected_spoke_edges.csv
diff --git a/tests/data/weighted edges/hubspoke_edges.csv → tests/data/weighted_edges/hubspoke_edges.csv b/tests/data/weighted edges/hubspoke_edges.csv → tests/data/weighted_edges/hubspoke_edges.csv
diff --git a/tests/data/weighted edges/line_edges.csv → tests/data/weighted_edges/line_edges.csv b/tests/data/weighted edges/line_edges.csv → tests/data/weighted_edges/line_edges.csv
diff --git a/...ted edges/mulithub_shared_spoke_edges.csv → ...ted_edges/mulithub_shared_spoke_edges.csv b/...ted edges/mulithub_shared_spoke_edges.csv → ...ted_edges/mulithub_shared_spoke_edges.csv
diff --git a/.../weighted edges/negative_cycles_edges.csv → .../weighted_edges/negative_cycles_edges.csv b/.../weighted edges/negative_cycles_edges.csv → .../weighted_edges/negative_cycles_edges.csv
diff --git a/tests/data/weighted edges/ring_edges.csv → tests/data/weighted_edges/ring_edges.csv b/tests/data/weighted edges/ring_edges.csv → tests/data/weighted_edges/ring_edges.csv
diff --git a/tests/data/weighted edges/tree_edges.csv → tests/data/weighted_edges/tree_edges.csv b/tests/data/weighted edges/tree_edges.csv → tests/data/weighted_edges/tree_edges.csv
diff --git a/tests/test/test_centrality.py b/tests/test/test_centrality.py
@@ -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)