test the sorting correctly (insertion order for the test graph was no…

…t sorted)
Pometry · Dec 13, 2024 · 03d03b0 · 03d03b0
1 parent 019f2b1
commit 03d03b0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tests/test_graphdb/test_graphdb.py b/python/tests/test_graphdb/test_graphdb.py
@@ -208,8 +208,10 @@ def test_edge_sorting():
 
     @with_disk_graph
     def check(g):
-        assert list(g.edges) == sorted(g.edges)
-        assert list(g.edges.explode()) == sorted(g.edges.explode())
+        assert sorted(g.edges, key=lambda e: e.id) == sorted(g.edges)
+        assert sorted(g.edges.explode(), key=lambda e: (*e.id, e.time)) == sorted(
+            g.edges.explode()
+        )
 
     check(g)