Touch up comments in test

Co-Authored-By: Jacob Sznajdman <breakanalysis@gmail.com>

Author: jjaderberg

alpha/alpha-embeddings/src/test/java/org/neo4j/gds/embeddings/node2vec/PositiveSampleProducerTest.java

@@ -71,14 +71,14 @@ void doesNotCauseStackOverflow() {
             10,
             TestProgressLogger.NULL_LOGGER
         );
-        // does not overflow the stack = passes test
 
         var counter = 0L;
         while (sampleProducer.hasNext()) {
             counter++;
             sampleProducer.next(new long[2]);
         }
 
+        // does not overflow the stack = passes test
         assertEquals(0, counter, "no samples possible here because walks are too short");
     }
 
@@ -93,7 +93,7 @@ void doesNotCauseStackOverflowDueToBadLuck() {
             .toArray(new long[][]{})
         );
 
-        // our sample producer is supposed to work within the first 3 walks
+        // our sample producer is supposed to work within the first nbrOfWalks - 1 walks
         // it prefetches in the constructor
         HugeDoubleArray probabilities = HugeDoubleArray.of(LongStream.range(0, nbrOfWalks).mapToDouble((l) -> 0).toArray());
         var sampleProducer = new PositiveSampleProducer(