From d39a58a9e0a7989e39ec86158dad5860dff01eed Mon Sep 17 00:00:00 2001 From: Galina Date: Thu, 30 Nov 2023 10:18:28 +0200 Subject: [PATCH] Fix e2e tests for explain --- tests/test_suite/run_test_command.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_suite/run_test_command.py b/tests/test_suite/run_test_command.py index 93bf558bf34..2375467399c 100644 --- a/tests/test_suite/run_test_command.py +++ b/tests/test_suite/run_test_command.py @@ -877,7 +877,7 @@ def otx_explain_testing(template, root, otx_dir, args, trained=False): assert os.path.exists(output_dir) if trained: assert len(os.listdir(output_dir)) > 0 - assert all([os.path.splitext(fname)[1] == ".tiff" for fname in os.listdir(output_dir)]) + assert all([os.path.splitext(fname)[1] in [".tiff", ".log"] for fname in os.listdir(output_dir)]) def otx_explain_testing_all_classes(template, root, otx_dir, args): @@ -964,7 +964,7 @@ def otx_explain_testing_process_saliency_maps(template, root, otx_dir, args, tra assert os.path.exists(output_dir) if trained: assert len(os.listdir(output_dir)) > 0 - assert all([os.path.splitext(fname)[1] == ".png" for fname in os.listdir(output_dir)]) + assert all([os.path.splitext(fname)[1] in [".png", ".log"] for fname in os.listdir(output_dir)]) def otx_explain_openvino_testing(template, root, otx_dir, args, trained=False): @@ -1005,7 +1005,7 @@ def otx_explain_openvino_testing(template, root, otx_dir, args, trained=False): assert os.path.exists(output_dir) if trained: assert len(os.listdir(output_dir)) > 0 - assert all([os.path.splitext(fname)[1] == ".tiff" for fname in os.listdir(output_dir)]) + assert all([os.path.splitext(fname)[1] in [".tiff", ".log"] for fname in os.listdir(output_dir)]) def otx_explain_all_classes_openvino_testing(template, root, otx_dir, args): @@ -1094,7 +1094,7 @@ def otx_explain_process_saliency_maps_openvino_testing(template, root, otx_dir, assert os.path.exists(output_dir) if trained: assert len(os.listdir(output_dir)) > 0 - assert all([os.path.splitext(fname)[1] == ".png" for fname in os.listdir(output_dir)]) + assert all([os.path.splitext(fname)[1] in [".png", ".log"] for fname in os.listdir(output_dir)]) def otx_find_testing():