From 81230388f1bd49f103856c7cfdb9e374c9a95a90 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Wed, 9 Nov 2016 12:50:28 -0500 Subject: Clean the file imports using a helper --- perf/stress_test.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) (limited to 'perf') diff --git a/perf/stress_test.py b/perf/stress_test.py index d1ee196b..f41f9f1e 100644 --- a/perf/stress_test.py +++ b/perf/stress_test.py @@ -35,10 +35,11 @@ def mk_main(file_count, call_count, line_count): return "\n".join(lines) - class StressTest(CoverageTest): def _compute_overhead(self, file_count, call_count, line_count): + self.clean_local_file_imports() + for idx in range(file_count): self.make_file('test{}.py'.format(idx), TEST_FILE) self.make_file('testmain.py', mk_main(file_count, call_count, line_count)) @@ -47,9 +48,7 @@ class StressTest(CoverageTest): self.import_local_file("testmain", None) baseline = time.perf_counter() - start - del sys.modules['testmain'] - for idx in range(file_count): - del sys.modules['test{}'.format(idx)] + self.clean_local_file_imports() start = time.perf_counter() cov = coverage.Coverage() -- cgit v1.2.1