From eea57d6019d7b13a3c9e01cf1ecb11b132e5e291 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Tue, 16 Jul 2024 16:33:26 +0200
Subject: Better print in perf.

---
 test/sca/perf_combine.py | 8 +++++++-
 test/sca/perf_zvp.py     | 9 ++++++++-
 2 files changed, 15 insertions(+), 2 deletions(-)

(limited to 'test/sca')

diff --git a/test/sca/perf_combine.py b/test/sca/perf_combine.py
index bd5db59..8ac274c 100644
--- a/test/sca/perf_combine.py
+++ b/test/sca/perf_combine.py
@@ -16,7 +16,13 @@ from pyecsca.sca import (
 
 
 @click.command()
-@click.option("-p", "--profiler", type=click.Choice(("py", "c", "raw")), default="py")
+@click.option(
+    "-p",
+    "--profiler",
+    type=click.Choice(("py", "c", "raw")),
+    default="py",
+    envvar="PROF",
+)
 @click.option("-o", "--operations", type=click.INT, default=100)
 @click.option(
     "-d",
diff --git a/test/sca/perf_zvp.py b/test/sca/perf_zvp.py
index 7a04dd6..34090df 100644
--- a/test/sca/perf_zvp.py
+++ b/test/sca/perf_zvp.py
@@ -11,12 +11,19 @@ from test.utils import Profiler
 
 
 @click.command()
-@click.option("-p", "--profiler", type=click.Choice(("py", "c", "raw")), default="py")
+@click.option(
+    "-p",
+    "--profiler",
+    type=click.Choice(("py", "c", "raw")),
+    default="py",
+    envvar="PROF",
+)
 @click.option(
     "-m",
     "--mod",
     type=click.Choice(("python", "gmp", "flint")),
     default="flint" if has_flint else "gmp" if has_gmp else "python",
+    envvar="MOD",
 )
 @click.option("-o", "--operations", type=click.INT, default=1)
 @click.option(
-- 
cgit v1.2.3-70-g09d2