Fix a completely unrelated knn test

neo4j · Nov 21, 2023 · 7228a19 · 7228a19
1 parent 5529293
commit 7228a19
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java b/algo/src/test/java/org/neo4j/gds/similarity/filteredknn/FilteredKnnIdMappingTest.java
@@ -74,7 +74,8 @@ void shouldIdMapTheSourceNodeFilter() {
         var sourceNodesInResult = result
             .similarityResultStream()
             .map(res -> res.node1)
+            .map(graph::toOriginalNodeId)
             .collect(Collectors.<Long>toSet());
-        assertThat(sourceNodesInResult).containsExactly(0L);
+        assertThat(sourceNodesInResult).containsExactly(lowestNeoId);
     }
 }