File tree Expand file tree Collapse file tree 1 file changed +7
-8
lines changed
proc/centrality/src/integrationTest/java/org/neo4j/gds/bridges Expand file tree Collapse file tree 1 file changed +7
-8
lines changed Original file line number Diff line number Diff 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 );
You can’t perform that action at this time.
0 commit comments