impl index spec

python/test_utils/filters_setup.py

@@ -284,3 +284,88 @@ def init_edges_graph2(graph):
         graph.edge(src, dst).add_constant_properties(props)
 
     return graph
+
+
+import tempfile
+from raphtory.graphql import GraphServer
+import json
+import re
+
+PORT = 1737
+
+
+def create_test_graph(g):
+    g.add_node(
+        1,
+        "a",
+        {
+            "prop1": 60,
+            "prop2": 31.3,
+            "prop3": "abc123",
+            "prop4": True,
+            "prop5": [1, 2, 3],
+        },
+        "fire_nation"
+    )
+    g.add_node(
+        1,
+        "b",
+        {"prop1": 10, "prop2": 31.3, "prop3": "abc223", "prop4": False},
+        "fire_nation"
+    )
+    g.add_node(
+        1,
+        "c",
+        {
+            "prop1": 20,
+            "prop2": 31.3,
+            "prop3": "abc333",
+            "prop4": True,
+            "prop5": [5, 6, 7],
+        },
+        "water_tribe"
+    )
+    g.add_node(
+        1,
+        "d",
+        {"prop1": 30, "prop2": 31.3, "prop3": "abc444", "prop4": False},
+        "air_nomads"
+    )
+    g.add_edge(
+        2,
+        "a",
+        "d",
+        {
+            "eprop1": 60,
+            "eprop2": 0.4,
+            "eprop3": "xyz123",
+            "eprop4": True,
+            "eprop5": [1, 2, 3],
+        },
+    )
+    g.add_edge(
+        2,
+        "b",
+        "d",
+        {
+            "eprop1": 10,
+            "eprop2": 1.7,
+            "eprop3": "xyz123",
+            "eprop4": True,
+            "eprop5": [3, 4, 5],
+        },
+    )
+    g.add_edge(
+        2,
+        "c",
+        "d",
+        {
+            "eprop1": 30,
+            "eprop2": 6.4,
+            "eprop3": "xyz123",
+            "eprop4": False,
+            "eprop5": [10],
+        },
+    )
+    return g
+

python/test_utils/utils.py

@@ -135,6 +135,24 @@ def run_graphql_error_test(query, expected_error_message, graph):
         ), f"Expected '{expected_error_message}', but got '{error_message}'"
 
 
+def run_graphql_error_test(query, expected_error_message, graph):
+    tmp_work_dir = tempfile.mkdtemp()
+    with GraphServer(tmp_work_dir).start(PORT) as server:
+        client = server.get_client()
+        client.send_graph(path="g", graph=graph)
+
+        with pytest.raises(Exception) as excinfo:
+            client.query(query)
+
+        full_error_message = str(excinfo.value)
+        match = re.search(r'"message":"(.*?)"', full_error_message)
+        error_message = match.group(1) if match else ""
+
+        assert (
+            error_message == expected_error_message
+        ), f"Expected '{expected_error_message}', but got '{error_message}'"
+
+
 def run_group_graphql_error_test(queries_and_expected_error_messages, graph):
     tmp_work_dir = tempfile.mkdtemp()
     with GraphServer(tmp_work_dir).start(PORT) as server:

python/tests/test_base_install/test_graphql/test_edge_sorting.py

@@ -1,7 +1,5 @@
 import pytest
-
 from raphtory import Graph, PersistentGraph
-
 from utils import run_graphql_test
 
 
@@ -67,12 +65,7 @@ def create_test_graph(g):
     return g
 
 
-EVENT_GRAPH = create_test_graph(Graph())
-
-PERSISTENT_GRAPH = create_test_graph(PersistentGraph())
-
-
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_nothing(graph):
     query = """
     query {
@@ -107,10 +100,11 @@ def test_graph_edge_sort_by_nothing(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_src(graph):
     query = """
     query {
@@ -145,10 +139,11 @@ def test_graph_edge_sort_by_src(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_dst(graph):
     query = """
     query {
@@ -183,10 +178,11 @@ def test_graph_edge_sort_by_dst(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_earliest_time(graph):
     query = """
     query {
@@ -221,10 +217,11 @@ def test_graph_edge_sort_by_earliest_time(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_earliest_time_reversed(graph):
     query = """
     query {
@@ -260,10 +257,11 @@ def test_graph_edge_sort_by_earliest_time_reversed(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph])
 def test_graph_edge_sort_by_latest_time(graph):
     query = """
     query {
@@ -298,10 +296,11 @@ def test_graph_edge_sort_by_latest_time(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [PersistentGraph])
 def test_graph_edge_sort_by_latest_time_persistent_graph(graph):
     query = """
     query {
@@ -337,10 +336,11 @@ def test_graph_edge_sort_by_latest_time_persistent_graph(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_eprop1(graph):
     query = """
     query {
@@ -375,10 +375,11 @@ def test_graph_edge_sort_by_eprop1(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_eprop2(graph):
     query = """
     query {
@@ -413,10 +414,11 @@ def test_graph_edge_sort_by_eprop2(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_eprop3(graph):
     query = """
     query {
@@ -451,10 +453,11 @@ def test_graph_edge_sort_by_eprop3(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_eprop4(graph):
     query = """
     query {
@@ -489,10 +492,11 @@ def test_graph_edge_sort_by_eprop4(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_eprop5(graph):
     query = """
     query {
@@ -527,10 +531,11 @@ def test_graph_edge_sort_by_eprop5(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_nonexistent_prop(graph):
     query = """
     query {
@@ -565,10 +570,11 @@ def test_graph_edge_sort_by_nonexistent_prop(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_combined(graph):
     query = """
     query {
@@ -603,10 +609,11 @@ def test_graph_edge_sort_by_combined(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)
 
 
-@pytest.mark.parametrize("graph", [EVENT_GRAPH, PERSISTENT_GRAPH])
+@pytest.mark.parametrize("graph", [Graph, PersistentGraph])
 def test_graph_edge_sort_by_combined_2(graph):
     query = """
     query {
@@ -641,4 +648,5 @@ def test_graph_edge_sort_by_combined_2(graph):
             }
         }
     }
+    graph = create_test_graph(graph())
     run_graphql_test(query, expected_output, graph)