Add a script to benchmark uninstalls (#78)

astral-sh · Oct 9, 2023 · f03c605 · f03c605
1 parent ba2b200
commit f03c605
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/scripts/benchmarks/uninstall.sh b/scripts/benchmarks/uninstall.sh
@@ -0,0 +1,17 @@
+#!/usr/bin/env sh
+
+###
+# Benchmark the uninstall command against `pip`.
+#
+# Example usage:
+#
+#   ./scripts/benchmarks/uninstall.sh numpy
+###
+
+set -euxo pipefail
+
+TARGET=${1}
+
+hyperfine --runs 20 --warmup 3 --prepare "rm -rf .venv && virtualenv .venv && source activate .venv/bin/activate && pip install ${TARGET}" \
+    "./target/release/puffin-cli uninstall ${TARGET}" \
+    "pip uninstall -y ${TARGET}"