{

# Function. Parameter 1 is the return code

if [[ $1 -ne "0" && $1 -ne "5" ]]; then

echo "Error: Test execution encountered exit code $1"

# as a bonus, make our script exit with the right error code.

exit $1

fi

{

echo "Checking Black Code Formatting"

black --check --exclude=".*\.ipynb" --extend-exclude=".venv|.pixi" --diff ./

check_errs $?

{

echo "Checking iSort Import Formatting"

isort --profile black --skip .venv --skip .pixi --check-only ./

check_errs $?

{

echo "Checking Missing Docstrings"

./docstring.py

check_errs $?

{

echo "Checking Flake8 Style Guide Enforcement"

flake8 --extend-exclude=.venv/,.pixi/ ./

check_errs $?

{

if [[ $print_mode == "verbose" ]]; then

if [[ `grep print */*.py | wc -l` -gt "0" ]]; then

echo "Error: print statement found in code"

grep print */*.py

exit 1

fi

fi

{

echo "Checking Missing fastmath flags in njit functions"

./fastmath.py --check stumpy

check_errs $?

echo "Checking hardcoded fastmath flags in njit functions"

if [[ $(grep -n fastmath= stumpy/*py | grep -vE 'fastmath=config' | wc -l) -gt "0" ]]; then

grep -n fastmath= stumpy/*py | grep -vE 'fastmath=config'

echo "Found one or more \`@njit()\` functions with a hardcoded \`fastmath\` flag."

exit 1

fi

{

echo "Checking Package Imports"

if [[ `grep '^from stumpy' stumpy/*py | wc -l` -gt "0" ]]; then

grep '^from stumpy' stumpy/*py

echo 'Error: please change "from stumpy" to "from ." '

exit 1

fi

{

# Check if there are any naive implementations not at start of test file

for testfile in tests/test_*.py

do

last_naive="$(grep -n 'def naive_' $testfile | tail -n 1 | awk -F: '{print $1}')"

first_test="$(grep -n 'def test_' $testfile | head -n 1 | awk -F: '{print $1}')"

if [[ ! -z $last_naive && ! -z $first_test && $last_naive -gt $first_test ]]; then

echo "Error: naive implementation found in the middle of $testfile line $last_naive"

exit 1

fi

done

{

if ! command -v ray &> /dev/null

then

echo "Ray Not Installed"

else

echo "Ray Installed"

fi

{

# Check if file does not exist OR file is empty

if [[ ! -e ".coveragerc_override" ]] || [[ ! -s ".coveragerc_override" ]]; then

echo "[report]" > .coveragerc_override

echo "; Regexes for lines to exclude from consideration" >> .coveragerc_override

echo "exclude_also =" >> .coveragerc_override

fi

{

# Generate a .coveragerc_override file that excludes Ray functions and tests

gen_coveragerc_boilerplate

echo " def .*_ray_*" >> .coveragerc_override

echo " def .*_ray\(*" >> .coveragerc_override

echo " def ray_.*" >> .coveragerc_override

echo " def test_.*_ray*" >> .coveragerc_override

{

if ! python -c "import pyfftw" &> /dev/null;

then

echo "pyFFTW cannot be imported"

else

echo "pyFFTW was successfully imported"

fi

{

gen_coveragerc_boilerplate

echo " def .*pyfftw.*" >> .coveragerc_override

echo " def test_.*pyfftw.*" >> .coveragerc_override

{

fcoveragerc=""

if ! command -v ray &> /dev/null;

then

echo "Ray not installed"

gen_ray_coveragerc

else

echo "Ray installed"

fi

if ! python -c "import pyfftw" &> /dev/null;

then

echo "pyFFTW cannot be imported"

gen_pyfftw_coveragerc

else

echo "pyFFTW was successfully imported"

fi

if [ -f .coveragerc_override ]; then

fcoveragerc="--rcfile=.coveragerc_override"

fi

{

set_coveragerc

coverage report --show-missing --fail-under=100 --skip-covered --omit=fastmath.py,docstring.py,versions.py $fcoveragerc

check_errs $?

{

# export NUMBA_DISABLE_JIT=1

# export NUMBA_ENABLE_CUDASIM=1

# Test one or more user-defined functions repeatedly

#

# ./test.sh custom tests/test_stump.py

# ./test.sh custom 5 tests/test_stump.py

# ./test.sh custom 5 tests/test_stump.py::test_stump_self_join

#

# You may mimic coverage testing conditions by disabling `numba` JIT

# and enabling the `cuda` simulator by setting two environment

# variables prior to calling `test.sh`:

#

# NUMBA_DISABLE_JIT=1 NUMBA_ENABLE_CUDASIM=1 ./test.sh custom 5 tests/test_gpu_stump.py

if [[ ${#custom_testfiles[@]} -eq "0" ]]; then

echo ""

echo "Error: Missing custom test file(s)"

echo "Please specify one or more custom test files"

echo "Example: ./test.sh custom tests/test_stump.py"

exit 1

else

for i in $(seq $max_iter)

do

echo "Custom Test: $i / $max_iter"

for testfile in "${custom_testfiles[@]}";

do

pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

done

clean_up

exit 0

fi

{

echo "Testing Numba JIT Compiled Functions"

SECONDS=0

if [[ ${#custom_testfiles[@]} -eq "0" ]]; then

for testfile in tests/test_*.py

do

pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

else

for testfile in "${custom_testfiles[@]}";

do

pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

fi

duration=$SECONDS

echo "Elapsed Time: $((duration / 60)) minutes and $((duration % 60)) seconds"

{

echo "Disabling Numba JIT and CUDA Compiled Functions"

export NUMBA_DISABLE_JIT=1

export NUMBA_ENABLE_CUDASIM=1

# echo "Testing Python Functions"

# pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning tests

# check_errs $?

echo "Testing Code Coverage"

coverage erase

# We always attempt to test everything but we may ignore things (ray, helper scripts) when we generate the coverage report

SECONDS=0

if [[ ${#custom_testfiles[@]} -eq "0" ]]; then

# Execute all tests

for testfile in tests/test_*.py;

do

coverage run --append --source=. -m pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

else

# Execute custom tests

for testfile in "${custom_testfiles[@]}";

do

coverage run --append --source=. -m pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

fi

duration=$SECONDS

echo "Elapsed Time: $((duration / 60)) minutes and $((duration % 60)) seconds"

show_coverage_report

{

echo "Testing Numba JIT CUDA GPU Compiled Functions"

#for testfile in tests/test_*gpu*.py tests/test_core.py tests/test_precision.py tests/test_non_normalized_decorator.py

for testfile in $(grep gpu tests/* | awk -v FS=':' '{print $1}' | uniq);

do

pytest -rsx -W ignore::RuntimeWarning -W ignore::DeprecationWarning -W ignore::UserWarning $testfile

check_errs $?

done

{

echo "Current working directory: " `pwd`

echo "Black version: " `python -c 'exec("try:\n\timport black;\n\tprint(black.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "Flake8 version: " `python -c 'exec("try:\n\timport flake8;\n\tprint(flake8.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "Python version: " `python -c "import platform; print(platform.python_version())"`

echo "NumPy version: " `python -c 'exec("try:\n\timport numpy;\n\tprint(numpy.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "SciPy version: " `python -c 'exec("try:\n\timport scipy;\n\tprint(scipy.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "Numba version: " `python -c 'exec("try:\n\timport numba;\n\tprint(numba.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "Dask version: " `python -c 'exec("try:\n\timport dask;\n\tprint(dask.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "Distributed version: " `python -c 'exec("try:\n\timport distributed;\n\tprint(distributed.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

echo "PyTest version: " `python -c 'exec("try:\n\timport pytest;\n\tprint(pytest.__version__);\nexcept ModuleNotFoundError:\n\tprint(\"Module Not Found\");")'`

exit 0

{

echo "Checking notebook links"

export JUPYTER_PLATFORM_DIRS=1

jupyter --paths

RERUNS=""

if [[ `pytest --trace-config | grep rerunfailures | wc -l` -gt "0" ]]; then

RERUNS="--reruns 5"

fi

pytest $RERUNS --check-links docs/Tutorial_*.ipynb notebooks/Tutorial_*.ipynb docs/*.md docs/*.rst ./*.md ./*.rst --check-links-ignore "https://dl.acm.org/doi/10.1145/3357223.3362721"

check_errs $?

{

test_count=$(pytest --collect-only -q | sed '$d' | sed '$d' | wc -l | sed 's/ //g')

echo "Found $test_count Unit Tests"

{

echo "Cleaning Up"

rm -rf "dask-worker-space"

rm -rf "stumpy/__pycache__/"

rm -rf "tests/__pycache__/"

rm -rf build dist stumpy.egg-info __pycache__

rm -f docs/*.nbconvert.ipynb

rm -rf ".coveragerc_override"

if [ -d "$site_pkgs/stumpy/__pycache__" ]; then

rm -rf $site_pkgs/stumpy/__pycache__/*nb*

fi

{

echo "testing notebooks"

for notebook in `grep ipynb docs/tutorials.rst | sed -e 's/^[ \t]*//'`

do

jupyter nbconvert --to notebook --execute "docs/$notebook"

check_errs $?

done