Skip to content

Commit 986f75e

Browse files
committed
Inline expected messages
1 parent 006826b commit 986f75e

File tree

1 file changed

+3
-6
lines changed

1 file changed

+3
-6
lines changed

algo/src/test/java/org/neo4j/gds/embeddings/graphsage/algo/GraphSageConfigTest.java

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -71,20 +71,17 @@ void specifyBatchesPerIteration(double samplingRatio, long nodeCount, int expect
7171

7272
@Test
7373
void shouldThrowIfNoPropertiesProvided() {
74-
var mapWrapper = CypherMapWrapper.create(Map.of("modelName", "foo"));
75-
var expectedMessage = "No value specified for the mandatory configuration parameter `featureProperties`";
76-
assertThatThrownBy(() -> GraphSageTrainConfig.of("", mapWrapper))
74+
assertThatThrownBy(() -> GraphSageTrainConfig.of("", CypherMapWrapper.create(Map.of("modelName", "foo"))))
7775
.isInstanceOf(IllegalArgumentException.class)
78-
.hasMessage(expectedMessage);
76+
.hasMessage("No value specified for the mandatory configuration parameter `featureProperties`");
7977
}
8078

8179
@Test
8280
void shouldThrowIfEmptyPropertiesProvided() {
8381
var mapWrapper = CypherMapWrapper.create(Map.of("modelName", "foo", "featureProperties", List.of()));
84-
var expectedMessage = "GraphSage requires at least one property.";
8582
assertThatThrownBy(() -> GraphSageTrainConfig.of("", mapWrapper))
8683
.isInstanceOf(IllegalArgumentException.class)
87-
.hasMessage(expectedMessage);
84+
.hasMessage("GraphSage requires at least one property.");
8885
}
8986

9087
@ParameterizedTest

0 commit comments

Comments
 (0)