Skip to content

Commit fd0a7c3

Browse files
committed
Add comprehensive GCN tests
1 parent 323ca81 commit fd0a7c3

3 files changed

Lines changed: 1083 additions & 1 deletion

File tree

geaflow/geaflow-dsl/geaflow-dsl-plan/src/test/java/org/apache/geaflow/dsl/GQLValidateGraphAlgorithmTest.java

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,5 +49,24 @@ public void testGraphAlgorithm() {
4949
.validate()
5050
.expectValidateType(
5151
"RecordType(BIGINT id, DOUBLE ARRAY embedding, BIGINT predicted_class, DOUBLE confidence)");
52+
53+
String script4 = "CALL GCN(2, 25) YIELD (id, embedding, predicted_class, confidence)\n"
54+
+ "RETURN id, embedding, predicted_class, confidence";
55+
56+
PlanTester.build()
57+
.gql(script4)
58+
.validate()
59+
.expectValidateType(
60+
"RecordType(BIGINT id, DOUBLE ARRAY embedding, BIGINT predicted_class, DOUBLE confidence)");
61+
62+
String script5 = "CALL GCN(2, 25, 'GCNTransFormFunction') "
63+
+ "YIELD (id, embedding, predicted_class, confidence)\n"
64+
+ "RETURN id, embedding, predicted_class, confidence";
65+
66+
PlanTester.build()
67+
.gql(script5)
68+
.validate()
69+
.expectValidateType(
70+
"RecordType(BIGINT id, DOUBLE ARRAY embedding, BIGINT predicted_class, DOUBLE confidence)");
5271
}
5372
}

0 commit comments

Comments
 (0)