From d031d80a615fc58e02edae06388620f257716d1a Mon Sep 17 00:00:00 2001
From: Remi Cresson <remi.cresson@inrae.fr>
Date: Wed, 23 Aug 2023 13:53:22 +0200
Subject: [PATCH] TEST: add testing for extra outputs
---
test/api_unittest.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/test/api_unittest.py b/test/api_unittest.py
index d16ea93..73b57b7 100644
--- a/test/api_unittest.py
+++ b/test/api_unittest.py
@@ -10,7 +10,8 @@ from otbtf.examples.tensorflow_v2x.fcnn import train_from_tfrecords
from otbtf.examples.tensorflow_v2x.fcnn.fcnn_model import INPUT_NAME, \
OUTPUT_SOFTMAX_NAME, OUTPUT_ARGMAX_NAME, TARGET_NAME
from otbtf.model import cropped_tensor_name
-from test_utils import resolve_paths, files_exist, run_command_and_compare
+from test_utils import resolve_paths, files_exist, run_command_and_compare, \
+ run_command_and_test_exist
INFERENCE_MAE_TOL = 10.0 # Dummy value: we don't really care of the mae value but rather the image size etc
--
GitLab