Skip to content

Commit 33db690

Browse files
Make BetweennessCentralityStreamProcTest.java order indifferent
1 parent c8ceddf commit 33db690

File tree

1 file changed

+23
-10
lines changed

1 file changed

+23
-10
lines changed

proc/centrality/src/test/java/org/neo4j/gds/betweenness/BetweennessCentralityStreamProcTest.java

Lines changed: 23 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
*/
2020
package org.neo4j.gds.betweenness;
2121

22+
import org.assertj.core.data.Offset;
2223
import org.junit.jupiter.api.BeforeEach;
2324
import org.junit.jupiter.api.Test;
2425
import org.neo4j.gds.BaseProcTest;
@@ -30,10 +31,11 @@
3031
import org.neo4j.gds.extension.Neo4jGraph;
3132
import org.neo4j.graphdb.QueryExecutionException;
3233

33-
import java.util.List;
3434
import java.util.Map;
3535

36+
import static org.assertj.core.api.Assertions.assertThat;
3637
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
38+
import static org.assertj.core.api.InstanceOfAssertFactories.DOUBLE;
3739

3840
class BetweennessCentralityStreamProcTest extends BaseProcTest {
3941

@@ -76,15 +78,26 @@ void testStream() {
7678
.streamMode()
7779
.yields();
7880

79-
assertCypherResult(
80-
query,
81-
List.of(
82-
Map.of("nodeId", idFunction.of("a"), "score", 0.0),
83-
Map.of("nodeId", idFunction.of("b"), "score", 3.0),
84-
Map.of("nodeId", idFunction.of("c"), "score", 4.0),
85-
Map.of("nodeId", idFunction.of("d"), "score", 3.0),
86-
Map.of("nodeId", idFunction.of("e"), "score", 0.0)
87-
));
81+
var expectedResultMap = Map.of(
82+
idFunction.of("a"), 0.0,
83+
idFunction.of("b"), 3.0,
84+
idFunction.of("c"), 4.0,
85+
idFunction.of("d"), 3.0,
86+
idFunction.of("e"), 0.0
87+
);
88+
89+
var rowCount = runQueryWithRowConsumer(query, (resultRow) -> {
90+
91+
var nodeId = resultRow.getNumber("nodeId");
92+
var expectedScore = expectedResultMap.get(nodeId);
93+
94+
assertThat(resultRow.getNumber("score")).asInstanceOf(DOUBLE).isCloseTo(
95+
expectedScore,
96+
Offset.offset(1e-6)
97+
);
98+
99+
});
100+
assertThat(rowCount).isEqualTo(5l);
88101
}
89102

90103
// FIXME: This should not be tested here

0 commit comments

Comments
 (0)