Reject --tracemalloc in pyperf command

psf · May 23, 2022 · 7363cea · 7363cea
1 parent 968f247
commit 7363cea
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.3.2
+-------------
+
+* Reject --tracemalloc in ``pyperf command``.
+  Patch by Yichen Yan
+
 Version 2.3.1 (2022-02-03)
 --------------------------
 

diff --git a/pyperf/_runner.py b/pyperf/_runner.py
@@ -305,6 +305,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: