tests.sh

set -e

# Test organ predictions
TotalSegmentator -i tests/reference_files/example_ct.nii.gz -o tests/unittest_prediction --test 1
pytest -v tests/test_end_to_end.py::test_end_to_end::test_prediction_liver
pytest -v tests/test_end_to_end.py::test_end_to_end::test_prediction_vertebrae

# Test multilabel prediction
TotalSegmentator -i tests/reference_files/example_ct.nii.gz -o tests/unittest_prediction.nii.gz --ml --test 1
pytest -v tests/test_end_to_end.py::test_end_to_end::test_prediction_multilabel

# Test organ predictions - fast
TotalSegmentator -i tests/reference_files/example_ct.nii.gz -o tests/unittest_prediction_fast --fast --statistics
pytest -v tests/test_end_to_end.py::test_end_to_end::test_prediction_liver_fast
pytest -v tests/test_end_to_end.py::test_end_to_end::test_prediction_vertebrae_fast
pytest -v tests/test_end_to_end.py::test_end_to_end::test_statistics

# Test vessel predictions
TotalSegmentator -i tests/reference_files/example_ct.nii.gz -o tests/unittest_prediction -ta lung_vessels --test 3
pytest -v tests/test_end_to_end.py::test_end_to_end::test_lung_vessels