Skip to content

Commit 5f00108

Browse files
committed
Beautify test setup
1 parent 69004ee commit 5f00108

File tree

1 file changed

+25
-24
lines changed

1 file changed

+25
-24
lines changed

alpha/alpha-proc/src/test/java/org/neo4j/gds/linkprediction/AdamicAdarProcTest.java

Lines changed: 25 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -41,23 +41,24 @@ protected void configuration(TestDatabaseManagementServiceBuilder builder) {
4141
}
4242

4343
private static final String DB_CYPHER =
44-
"CREATE (mark:Person {name: 'Mark'})\n" +
45-
"CREATE (michael:Person {name: 'Michael'})\n" +
46-
"CREATE (praveena:Person {name: 'Praveena'})\n" +
47-
"CREATE (ryan:Person {name: 'Ryan'})\n" +
48-
"CREATE (karin:Person {name: 'Karin'})\n" +
49-
"CREATE (jennifer:Person {name: 'Jennifer'})\n" +
50-
"CREATE (elaine:Person {name: 'Elaine'})\n" +
44+
"CREATE " +
45+
" (mark:Person {name: 'Mark'}), " +
46+
" (michael:Person {name: 'Michael'}), " +
47+
" (praveena:Person {name: 'Praveena'}), " +
48+
" (ryan:Person {name: 'Ryan'}), " +
49+
" (karin:Person {name: 'Karin'}), " +
50+
" (jennifer:Person {name: 'Jennifer'}), " +
51+
" (elaine:Person {name: 'Elaine'}), " +
5152

52-
"MERGE (jennifer)-[:FRIENDS]-(ryan)\n" +
53-
"MERGE (jennifer)-[:FRIENDS]-(karin)\n" +
54-
"MERGE (elaine)-[:FRIENDS]-(ryan)\n" +
55-
"MERGE (elaine)-[:FRIENDS]-(karin)\n" +
53+
" (jennifer)-[:FRIENDS]->(ryan), " +
54+
" (jennifer)-[:FRIENDS]->(karin), " +
55+
" (elaine)-[:FRIENDS]->(ryan), " +
56+
" (elaine)-[:FRIENDS]->(karin), " +
5657

57-
"MERGE (mark)-[:FRIENDS]-(michael)\n" +
58-
"MERGE (mark)-[:WORKS_WITH]->(michael)\n" +
58+
" (mark)-[:FRIENDS]->(michael), " +
59+
" (mark)-[:WORKS_WITH]->(michael), " +
5960

60-
"MERGE (praveena)-[:FRIENDS]->(michael)";
61+
" (praveena)-[:FRIENDS]->(michael)";
6162

6263
@BeforeEach
6364
void setUp() throws Exception {
@@ -68,8 +69,8 @@ void setUp() throws Exception {
6869
@Test
6970
void oneNodeInCommon() {
7071
String controlQuery =
71-
"MATCH (p1:Person {name: 'Mark'})\n" +
72-
"MATCH (p2:Person {name: 'Praveena'})\n" +
72+
"MATCH (p1:Person {name: 'Mark'}) " +
73+
"MATCH (p2:Person {name: 'Praveena'}) " +
7374
"RETURN gds.alpha.linkprediction.adamicAdar(p1, p2) AS score, " +
7475
" 1/log(3) AS cypherScore";
7576

@@ -80,8 +81,8 @@ void oneNodeInCommon() {
8081
@Test
8182
void oneNodeInCommonExplicit() {
8283
String controlQuery =
83-
"MATCH (p1:Person {name: 'Mark'})\n" +
84-
"MATCH (p2:Person {name: 'Praveena'})\n" +
84+
"MATCH (p1:Person {name: 'Mark'}) " +
85+
"MATCH (p2:Person {name: 'Praveena'}) " +
8586
"RETURN gds.alpha.linkprediction.adamicAdar(p1, p2, " +
8687
"{relationshipQuery: 'FRIENDS', direction: 'BOTH'}) AS score," +
8788
"1/log(2) AS cypherScore";
@@ -93,8 +94,8 @@ void oneNodeInCommonExplicit() {
9394
@Test
9495
void twoNodesInCommon() {
9596
String controlQuery =
96-
"MATCH (p1:Person {name: 'Jennifer'})\n" +
97-
"MATCH (p2:Person {name: 'Elaine'})\n" +
97+
"MATCH (p1:Person {name: 'Jennifer'}) " +
98+
"MATCH (p2:Person {name: 'Elaine'}) " +
9899
"RETURN gds.alpha.linkprediction.adamicAdar(p1, p2) AS score, " +
99100
" 1/log(2) + 1/log(2) AS cypherScore";
100101

@@ -105,8 +106,8 @@ void twoNodesInCommon() {
105106
@Test
106107
void noNeighbors() {
107108
String controlQuery =
108-
"MATCH (p1:Person {name: 'Jennifer'})\n" +
109-
"MATCH (p2:Person {name: 'Ryan'})\n" +
109+
"MATCH (p1:Person {name: 'Jennifer'}) " +
110+
"MATCH (p2:Person {name: 'Ryan'}) " +
110111
"RETURN gds.alpha.linkprediction.adamicAdar(p1, p2) AS score, " +
111112
" 0.0 AS cypherScore";
112113

@@ -117,8 +118,8 @@ void noNeighbors() {
117118
@Test
118119
void bothNodesTheSame() {
119120
String controlQuery =
120-
"MATCH (p1:Person {name: 'Praveena'})\n" +
121-
"MATCH (p2:Person {name: 'Praveena'})\n" +
121+
"MATCH (p1:Person {name: 'Praveena'}) " +
122+
"MATCH (p2:Person {name: 'Praveena'}) " +
122123
"RETURN gds.alpha.linkprediction.adamicAdar(p1, p2) AS score, " +
123124
" 0.0 AS cypherScore";
124125

0 commit comments

Comments
 (0)