Skip to content
Snippets Groups Projects
Commit 4ccefa94 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

[test] Check saliency outputs in experiment test

parent 446eebf9
No related branches found
No related tags found
2 merge requests!18Update tests,!16Make square centre-padding a model transform
......@@ -498,6 +498,38 @@ def test_experiment(temporary_basedir):
assert os.path.exists(
os.path.join(output_folder, "evaluation", "summary.rst")
)
assert os.path.exists(os.path.join(output_folder, "gradcam", "saliencies"))
assert (
len(
glob.glob(
os.path.join(
output_folder,
"gradcam",
"saliencies",
"CXR_png",
"MCUCXR_*.npy",
)
)
)
== 138
)
assert os.path.exists(
os.path.join(output_folder, "gradcam", "visualizations")
)
assert (
len(
glob.glob(
os.path.join(
output_folder,
"gradcam",
"visualizations",
"CXR_png",
"MCUCXR_*.png",
)
)
)
== 58
)
# This script does not work anymore, either fix or remove the script + this test
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment