1#!/usr/bin/env bash
2
3set -o errexit
4set -o nounset
5
6function run_python_tests() {
7    GIT_ROOT=$(git rev-parse --show-toplevel)
8  
9    pushd "$GIT_ROOT" >/dev/null
10      # If there's a `scripts` folder with a test script, run that
11      if find "scripts" -name 'run_*_tests.sh' >/dev/null; then
12          print_info "-> bash scripts/run_*_tests.sh"
13          bash scripts/run_*_tests.sh
14          return 0
15      fi
16    popd >/dev/null
17  
18    REPO_NAME="$(basename $(pwd))"
19
20    # Run ruff to do Python formatting
21    print_info "-> ruff format"
22    ruff format .
23    
24    echo ""
25    
26    print_info "-> ruff check --fix"
27    ruff check --fix .
28
29    echo ""
30
31    # This is one repo which is a bit special -- I'm gradually trying to chase
32    # the code into `src`, but for now I have to remember to look at *.py
33    if [[ "$REPO_NAME" = "library-lookup" ]]
34    then
35        print_info '-> mypy *.py src tests'
36        mypy *.py src tests
37    else
38        print_info '-> mypy src tests'
39
40        if mypy src tests >/dev/null
41        then
42            mypy src tests --no-color-output
43        else
44            mypy src tests
45        fi
46    fi
47
48    echo ""
49
50    # Run the tests.
51    print_info "-> coverage run -m pytest tests"
52    coverage run -m pytest tests --quiet
53
54    echo ""
55
56    print_info "-> coverage report"
57
58    if [[ $(coverage report --format=total) = "100" ]]
59    then
60        echo "100% coverage!"
61    else
62        coverage report
63    fi
64}
65
66function run_rust_tests() {
67    print_info "-> cargo fmt"
68    cargo fmt
69
70    echo ""
71
72    print_info "-> cargo build"
73    cargo build
74
75    echo ""
76
77    print_info -n "-> cargo test"
78    cargo test --quiet
79}
80
81if test -f Cargo.toml
82then
83    run_rust_tests
84else
85    run_python_tests "$@"
86fi