Skip to content

Commit 803e143

Browse files
committed
Fix tests and add unsupported APSP estimate for sessions
1 parent 0046dfc commit 803e143

File tree

1 file changed

+9
-10
lines changed

1 file changed

+9
-10
lines changed

algo/src/test/java/org/neo4j/gds/allshortestpaths/AllShortestPathsMemoryEstimateDefinitionTest.java

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,6 @@
2121

2222
import org.junit.jupiter.params.ParameterizedTest;
2323
import org.junit.jupiter.params.provider.CsvSource;
24-
import org.neo4j.gds.assertions.MemoryEstimationAssert;
2524
import org.neo4j.gds.core.concurrency.Concurrency;
2625

2726
import static org.neo4j.gds.assertions.MemoryEstimationAssert.assertThat;
@@ -30,19 +29,19 @@ class AllShortestPathsMemoryEstimateDefinitionTest {
3029

3130
@ParameterizedTest
3231
@CsvSource({
33-
"10_000, 1, false, 280",
34-
"10_000, 4, false, 544",
35-
"500_000, 4, false, 27_200",
36-
"10_000_000, 4, false, 544_000",
37-
"10_000, 1, true, 1_120_456",
38-
"10_000, 4, true, 1_120_720",
39-
"500_000, 4, true, 56_000_720",
40-
"10_000_000, 4, true, 1_120_000_720"
32+
"10_000, 1, false, 320",
33+
"10_000, 4, false, 584",
34+
"500_000, 4, false, 584",
35+
"10_000_000, 4, false, 584",
36+
"10_000, 1, true, 120_280",
37+
"10_000, 4, true, 480_976",
38+
"500_000, 4, true, 24_000_976",
39+
"10_000_000, 4, true, 480_000_976"
4140
})
4241
void testMemoryEstimation(long nodeCount, int concurrency, boolean weighted, long expectedMemory) {
4342
var memoryEstimation = new AllShortestPathsMemoryEstimateDefinition(weighted).memoryEstimation();
4443
assertThat(memoryEstimation)
4544
.memoryRange(nodeCount, new Concurrency(concurrency))
4645
.hasSameMinAndMaxEqualTo(expectedMemory);
4746
}
48-
}
47+
}

0 commit comments

Comments
 (0)