Skip to content

Commit caf819e

Browse files
committed
migrate estimation cli leiden to application layer
1 parent baeff69 commit caf819e

File tree

9 files changed

+61
-376
lines changed

9 files changed

+61
-376
lines changed

algo/src/main/java/org/neo4j/gds/leiden/LeidenAlgorithmFactory.java

Lines changed: 0 additions & 96 deletions
This file was deleted.

algo/src/test/java/org/neo4j/gds/leiden/LeidenAlgorithmFactoryTest.java

Lines changed: 10 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -20,14 +20,15 @@
2020
package org.neo4j.gds.leiden;
2121

2222
import org.junit.jupiter.api.Test;
23-
import org.neo4j.gds.core.utils.progress.tasks.ProgressTracker;
23+
import org.neo4j.gds.applications.algorithms.community.CommunityAlgorithms;
24+
import org.neo4j.gds.applications.algorithms.community.LeidenTask;
2425
import org.neo4j.gds.core.utils.progress.tasks.Tasks;
2526
import org.neo4j.gds.gdl.GdlFactory;
2627

2728
import java.util.List;
2829

2930
import static org.assertj.core.api.Assertions.assertThat;
30-
import static org.assertj.core.api.Assertions.assertThatThrownBy;
31+
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
3132

3233
class LeidenAlgorithmFactoryTest {
3334

@@ -38,8 +39,7 @@ void shouldProduceProgressTask() {
3839

3940
var graph = GdlFactory.of(" CREATE (a:NODE), (b:NODE) ").build().getUnion();
4041

41-
var task = new LeidenAlgorithmFactory<>().progressTask(graph, config);
42-
42+
var task = LeidenTask.create(graph, config);
4343
var initialization = Tasks.leaf("Initialization", 2);
4444

4545
var iteration = Tasks.iterativeDynamic("Iteration", () ->
@@ -59,13 +59,11 @@ void shouldProduceProgressTask() {
5959
@Test
6060
void shouldThrowIfNotUndirected() {
6161
var graph = GdlFactory.of("(a)-->(b)").build().getUnion();
62-
var config = LeidenStatsConfigImpl.builder().maxLevels(3).build();
63-
var leidenFactory = new LeidenAlgorithmFactory<>();
64-
assertThatThrownBy(() -> leidenFactory.build(
65-
graph,
66-
config,
67-
ProgressTracker.NULL_TRACKER
68-
)).hasMessageContaining(
69-
"undirected");
62+
63+
var communityAlgorithms = new CommunityAlgorithms(null, null);
64+
65+
assertThatIllegalArgumentException()
66+
.isThrownBy(() -> communityAlgorithms.leiden(graph, null))
67+
.withMessage("The Leiden algorithm works only with undirected graphs. Please orient the edges properly");
7068
}
7169
}

algorithm-specifications/src/main/java/org/neo4j/gds/leiden/Constants.java

Lines changed: 0 additions & 27 deletions
This file was deleted.

algorithm-specifications/src/main/java/org/neo4j/gds/leiden/LeidenMutateSpec.java

Lines changed: 0 additions & 57 deletions
This file was deleted.

algorithm-specifications/src/main/java/org/neo4j/gds/leiden/LeidenStatsSpec.java

Lines changed: 0 additions & 56 deletions
This file was deleted.

algorithm-specifications/src/main/java/org/neo4j/gds/leiden/LeidenStreamSpec.java

Lines changed: 0 additions & 56 deletions
This file was deleted.

0 commit comments

Comments
 (0)