Sandbox::https://github.com/vortex-data/vortex/blob/develop/vortex-python/run

Repository navigation

executable file

28 lines (21 loc) · 986 Bytes

#!/bin/bash
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright the Vortex contributors
# Run Python benchmarks with a release-profile build of the native extension.
#   ./run_benchmarks.sh                     # run all benchmarks
#   ./run_benchmarks.sh -k "test_scan"      # run benchmarks matching a pattern
#   ./run_benchmarks.sh --benchmark-only    # skip non-benchmark tests (if any)
# All arguments are forwarded to pytest.
set -ex -o pipefail
ROOT=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )/.." &> /dev/null && pwd )
PYTHON_DIR="$ROOT/vortex-python"
# Ensure all packages are synced (includes pytest-benchmark dev dependency).
uv sync --all-packages
source "$ROOT/.venv/bin/activate"
# Build the native extension in release mode so benchmarks reflect production performance.
maturin develop --release --manifest-path "$PYTHON_DIR/Cargo.toml"
# Run benchmarks. Extra args are forwarded to pytest.
pytest "$PYTHON_DIR/benchmark" "$@"

FilesExpand file tree

run_benchmarks.sh

Latest commit

History

run_benchmarks.sh

File metadata and controls