From 34f8c11c57f7f2899f8c163e37c36e418261ca6d Mon Sep 17 00:00:00 2001
From: Harald Musum <musum@yahooinc.com>
Date: Mon, 14 Feb 2022 16:04:58 +0100
Subject: Skip tests if onnxruntime unavailable, part 3

---
 .../com/yahoo/application/container/ContainerModelEvaluationTest.java  | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'application/src/test')

diff --git a/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java b/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java
index 193fec3f7bb..f76008b4e02 100644
--- a/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java
+++ b/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.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.application.container;
 
+import ai.vespa.modelintegration.evaluator.OnnxEvaluator;
 import com.yahoo.application.Application;
 import com.yahoo.application.Networking;
 import com.yahoo.application.container.handler.Request;
@@ -17,6 +18,7 @@ import java.nio.charset.CharacterCodingException;
 import java.nio.charset.StandardCharsets;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assume.assumeTrue;
 
 /**
  * Verify that we can create a JDisc (and hence Application) instance capable of doing model evaluation
@@ -37,6 +39,7 @@ public class ContainerModelEvaluationTest {
 
     @Test
     public void testCreateApplicationInstanceWithModelEvaluation() {
+        assumeTrue(OnnxEvaluator.isRuntimeAvailable());
         try (Application application =
                      Application.fromApplicationPackage(new File("src/test/app-packages/model-evaluation"),
                                                         Networking.disable)) {
-- 
cgit v1.2.3