Stronger testing

Co-authored-by: Mats Rydberg <mats@neotechnology.com>

Author: IoannisPanagiotas

proc/path-finding/src/test/java/org/neo4j/gds/paths/spanningtree/SpanningTreeMutateProcTest.java

@@ -101,7 +101,8 @@ void testYields(String tieredProcedure) {
                 "computeMillis",
                 "mutateMillis",
                 "effectiveNodeCount",
-                "relationshipsWritten"
+                "relationshipsWritten",
+                "totalWeight"
             );
 
         var graph = findLoadedGraph(DEFAULT_GRAPH_NAME);
@@ -116,6 +117,7 @@ void testYields(String tieredProcedure) {
                 assertThat(res.getNumber("mutateMillis").longValue()).isGreaterThanOrEqualTo(0L);
                 assertThat(res.getNumber("effectiveNodeCount").longValue()).isEqualTo(5L);
                 assertThat(res.getNumber("relationshipsWritten").longValue()).isEqualTo(4L);
+                assertThat(res.getNumber("totalWeight").doubleValue()).isEqualTo(12.0);
 
             }
         );

proc/path-finding/src/test/java/org/neo4j/gds/paths/spanningtree/SpanningTreeWriteProcTest.java

@@ -134,7 +134,8 @@ void testYieldsWithBeta() {
                 "computeMillis",
                 "writeMillis",
                 "effectiveNodeCount",
-                "relationshipsWritten"
+                "relationshipsWritten",
+                "totalWeight"
             );
 
         runQueryWithRowConsumer(
@@ -145,6 +146,8 @@ void testYieldsWithBeta() {
                 assertThat(res.getNumber("writeMillis").longValue()).isGreaterThanOrEqualTo(0L);
                 assertThat(res.getNumber("effectiveNodeCount").longValue()).isEqualTo(5L);
                 assertThat(res.getNumber("relationshipsWritten").longValue()).isEqualTo(4L);
+                assertThat(res.getNumber("totalWeight").doubleValue()).isEqualTo(12.0);
+
 
             }
         );