Reject --tracemalloc in pyperf command (psf#131)

oraluben · web-flow · commit 1fd8dae4b621 · 2022-07-21T10:30:51.000+09:00
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:
