# or whatever path is set as $TF_WHEEL.

setup_file() {
    cd /tf/pkg
    if [[ -z "$TF_WHEEL" ]]; then
        export TF_WHEEL=$(find /tf/pkg -iname "*.whl")
    fi
}

teardown_file() {
    rm -rf /tf/venv
}

@test "Wheel is manylinux2014 (manylinux_2_17) compliant" {
    python3 -m auditwheel show "$TF_WHEEL" > audit.txt

# TODO(b/279852433) Replace deps(//tensorflow/...) with deps(//...)
@test "Verify that it's possible to query every TensorFlow target without BUILD errors" {
    bazel query "deps(//tensorflow/...)" > /dev/null
}

teardown_file() {
    bazel shutdown