Skip to content

Commit 22c1319

Browse files
committed
Fix projection in GraphStoreExportProcTest
1 parent 14e7a2a commit 22c1319

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

proc/catalog/src/test/java/org/neo4j/graphalgo/catalog/GraphStoreExportProcTest.java

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,10 @@
2323
import org.junit.jupiter.api.Test;
2424
import org.neo4j.graphalgo.BaseProcTest;
2525
import org.neo4j.graphalgo.GdsCypher;
26+
import org.neo4j.graphalgo.Orientation;
27+
import org.neo4j.graphalgo.PropertyMapping;
28+
import org.neo4j.graphalgo.PropertyMappings;
29+
import org.neo4j.graphalgo.RelationshipProjection;
2630

2731
import static org.hamcrest.MatcherAssert.assertThat;
2832
import static org.hamcrest.Matchers.greaterThan;
@@ -56,12 +60,18 @@ void exportGraph() {
5660
.withAnyLabel()
5761
.withNodeProperty("prop1")
5862
.withNodeProperty("prop2")
59-
.withRelationshipType("REL1")
60-
.withRelationshipType("REL2")
61-
.withRelationshipType("REL3")
62-
.withRelationshipProperty("weight1")
63-
.withRelationshipProperty("weight2")
64-
.withRelationshipProperty("weight3")
63+
.withRelationshipType("REL1", RelationshipProjection
64+
.of("REL1", Orientation.NATURAL)
65+
.withProperties(PropertyMappings.of(PropertyMapping.of("weight1")))
66+
)
67+
.withRelationshipType("REL2", RelationshipProjection
68+
.of("REL2", Orientation.NATURAL)
69+
.withProperties(PropertyMappings.of(PropertyMapping.of("weight2")))
70+
)
71+
.withRelationshipType("REL3", RelationshipProjection
72+
.of("REL3", Orientation.NATURAL)
73+
.withProperties(PropertyMappings.of(PropertyMapping.of("weight3")))
74+
)
6575
.graphCreate("test-graph")
6676
.yields());
6777

0 commit comments

Comments
 (0)