2121
2222import org .junit .jupiter .params .ParameterizedTest ;
2323import org .junit .jupiter .params .provider .CsvSource ;
24- import org .neo4j .gds .assertions .MemoryEstimationAssert ;
2524import org .neo4j .gds .core .concurrency .Concurrency ;
2625
2726import 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