summaryrefslogtreecommitdiff
path: root/tests/test_cmdline.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_cmdline.py')
-rw-r--r--tests/test_cmdline.py8
1 files changed, 2 insertions, 6 deletions
diff --git a/tests/test_cmdline.py b/tests/test_cmdline.py
index 116ad7d3..223e1313 100644
--- a/tests/test_cmdline.py
+++ b/tests/test_cmdline.py
@@ -693,13 +693,9 @@ class CmdMainTest(CoverageTest):
def setUp(self):
super(CmdMainTest, self).setUp()
- self.old_CoverageScript = coverage.cmdline.CoverageScript
+ old_CoverageScript = coverage.cmdline.CoverageScript
coverage.cmdline.CoverageScript = self.CoverageScriptStub
- self.addCleanup(self.cleanup_coverage_script)
-
- def cleanup_coverage_script(self):
- """Restore CoverageScript when the test is done."""
- coverage.cmdline.CoverageScript = self.old_CoverageScript
+ self.addCleanup(setattr, coverage.cmdline, 'CoverageScript', old_CoverageScript)
def test_normal(self):
ret = coverage.cmdline.main(['hello'])