Skip to content

Commit 1c084ec

Browse files
Fix e2e tests for explain (#2681)
1 parent de0f5ae commit 1c084ec

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/test_suite/run_test_command.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -877,7 +877,7 @@ def otx_explain_testing(template, root, otx_dir, args, trained=False):
877877
assert os.path.exists(output_dir)
878878
if trained:
879879
assert len(os.listdir(output_dir)) > 0
880-
assert all([os.path.splitext(fname)[1] == ".tiff" for fname in os.listdir(output_dir)])
880+
assert all([os.path.splitext(fname)[1] in [".tiff", ".log"] for fname in os.listdir(output_dir)])
881881

882882

883883
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
964964
assert os.path.exists(output_dir)
965965
if trained:
966966
assert len(os.listdir(output_dir)) > 0
967-
assert all([os.path.splitext(fname)[1] == ".png" for fname in os.listdir(output_dir)])
967+
assert all([os.path.splitext(fname)[1] in [".png", ".log"] for fname in os.listdir(output_dir)])
968968

969969

970970
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):
10051005
assert os.path.exists(output_dir)
10061006
if trained:
10071007
assert len(os.listdir(output_dir)) > 0
1008-
assert all([os.path.splitext(fname)[1] == ".tiff" for fname in os.listdir(output_dir)])
1008+
assert all([os.path.splitext(fname)[1] in [".tiff", ".log"] for fname in os.listdir(output_dir)])
10091009

10101010

10111011
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,
10941094
assert os.path.exists(output_dir)
10951095
if trained:
10961096
assert len(os.listdir(output_dir)) > 0
1097-
assert all([os.path.splitext(fname)[1] == ".png" for fname in os.listdir(output_dir)])
1097+
assert all([os.path.splitext(fname)[1] in [".png", ".log"] for fname in os.listdir(output_dir)])
10981098

10991099

11001100
def otx_find_testing():

0 commit comments

Comments
 (0)