diff options
Diffstat (limited to 'config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java')
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java b/config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java index 556d91acb70..bd07513b704 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/ml/ImportedModelTester.java @@ -1,6 +1,7 @@ // Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. package com.yahoo.vespa.model.ml; +import com.yahoo.config.FileReference; import com.yahoo.config.model.ApplicationPackageTester; import ai.vespa.rankingexpression.importer.configmodelview.MlModelImporter; import com.yahoo.config.model.deploy.DeployState; @@ -22,9 +23,9 @@ import java.io.UncheckedIOException; import java.util.List; import java.util.Optional; -import static junit.framework.TestCase.assertTrue; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; +import static org.junit.Assert.assertTrue; /** * Helper for testing of imported models. @@ -74,6 +75,8 @@ public class ImportedModelTester { assertEquals(constantName, constant.getName()); assertTrue(constant.getFileName().endsWith(constantApplicationPackagePath.toString())); + assertTrue(model.fileReferences().contains(new FileReference(constant.getFileName()))); + if (expectedSize.isPresent()) { Path constantPath = applicationDir.append(constantApplicationPackagePath); assertTrue("Constant file '" + constantPath + "' has been written", |