Skip to content

Commit c3b6ec8

Browse files
unflaeky test
1 parent ab32168 commit c3b6ec8

File tree

1 file changed

+7
-8
lines changed

1 file changed

+7
-8
lines changed

proc/centrality/src/integrationTest/java/org/neo4j/gds/bridges/BridgesStreamProcTest.java

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -72,16 +72,15 @@ void shouldStreamBackResults(){
7272
.streamMode()
7373
.yields();
7474

75-
var expectedFrom = idFunction.of("a");
76-
var expectedTo = idFunction.of("b");
75+
var expectedNode1 = idFunction.of("a");
76+
var expectedNode2 = idFunction.of("b");
7777

7878
var rowCount = runQueryWithRowConsumer(query, (resultRow) -> {
79-
80-
var fromId = resultRow.getNumber("from");
81-
var toId = resultRow.getNumber("to");
82-
83-
assertThat(fromId.longValue()).isEqualTo(expectedFrom);
84-
assertThat(toId.longValue()).isEqualTo(expectedTo);
79+
var fromId = resultRow.getNumber("from");
80+
var toId = resultRow.getNumber("to");
81+
assertThat(fromId.longValue()).isNotEqualTo(toId.longValue());
82+
assertThat(fromId.longValue()).isIn(expectedNode1, expectedNode2);
83+
assertThat(toId.longValue()).isIn(expectedNode1, expectedNode2);
8584

8685
});
8786
assertThat(rowCount).isEqualTo(1l);

0 commit comments

Comments
 (0)