Reject --tracemalloc in pyperf command (#131)

psf · Jul 21, 2022 · 1fd8dae · 1fd8dae
1 parent 6be960e
commit 1fd8dae
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/TODO.rst b/TODO.rst
@@ -2,7 +2,6 @@ BUGS
 ====
 
 * Repair PyPy on Travis
-* Reject --tracemalloc in bench_command()
 * BUG: "python pyperf timeit --compare-to=pypy" doesn't tune correctly the runner
   for pypy. pypy requires more warmup values (10) than cpython (1).
 * BUG: --duplicate of timeit must be ignored in PyPy, see the discussion

diff --git a/doc/changelog.rst b/doc/changelog.rst
@@ -1,6 +1,12 @@
 Changelog
 =========
 
+Version 2.4.1
+-------------
+
+* Reject --tracemalloc in ``pyperf command``.
+  Patch by Yichen Yan
+
 Version 2.4.0 (2022-07-20)
 --------------------------
 

diff --git a/pyperf/_runner.py b/pyperf/_runner.py
@@ -324,6 +324,8 @@ def _process_args_impl(self):
             self._only_in_worker("--worker-task")
 
         if args.tracemalloc:
+            if args.action == 'command':
+                raise CLIError('--tracemalloc cannot be used with pyperf command')
             try:
                 import tracemalloc   # noqa
             except ImportError as exc: