diff options
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java b/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java index 509d8527bf5..e7f8086e554 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/container/xml/EmbedderTestCase.java @@ -24,6 +24,7 @@ import java.nio.charset.StandardCharsets; import java.util.List; import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertTrue; import static org.junit.jupiter.api.Assertions.fail; public class EmbedderTestCase { @@ -91,8 +92,7 @@ public class EmbedderTestCase { " </config>" + "</component>"; assertTransformThrows(embedder, - "Unknown model id 'my_model_id' on 'transformerModel'. " + - "Available models are [bert-base-uncased, minilm-l6-v2, mpnet-base-v2]", + "Unknown model id 'my_model_id' on 'transformerModel'", true); } @@ -194,7 +194,7 @@ public class EmbedderTestCase { ModelIdResolver.resolveModelIds(createElement(embedder), hosted); fail("Expected exception was not thrown: " + expectedMessage); } catch (IllegalArgumentException e) { - assertEquals(expectedMessage, e.getMessage()); + assertTrue(e.getMessage().contains(expectedMessage), "Expected error message not found"); } } |