From 34cd2ae69fc9a004c95e6c361aa7cd3ae4db2caa Mon Sep 17 00:00:00 2001 From: "Gregory P. Smith" Date: Thu, 22 Jan 2015 14:38:00 -0800 Subject: Break up TestCommandLine.test_env_var into four distinct tests. --- Lib/test/test_tracemalloc.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'Lib/test/test_tracemalloc.py') diff --git a/Lib/test/test_tracemalloc.py b/Lib/test/test_tracemalloc.py index 33b0dc2b0e..bc22450f0f 100644 --- a/Lib/test/test_tracemalloc.py +++ b/Lib/test/test_tracemalloc.py @@ -748,26 +748,28 @@ class TestFilters(unittest.TestCase): class TestCommandLine(unittest.TestCase): - def test_env_var(self): + def test_env_var_disabled_by_default(self): # not tracing by default code = 'import tracemalloc; print(tracemalloc.is_tracing())' ok, stdout, stderr = assert_python_ok('-c', code) stdout = stdout.rstrip() self.assertEqual(stdout, b'False') - # PYTHON* environment variables must be ignored when -E option is - # present + def test_env_var_ignored_with_E(self): + """PYTHON* environment variables must be ignored when -E is present.""" code = 'import tracemalloc; print(tracemalloc.is_tracing())' ok, stdout, stderr = assert_python_ok('-E', '-c', code, PYTHONTRACEMALLOC='1') stdout = stdout.rstrip() self.assertEqual(stdout, b'False') + def test_env_var_enabled_at_startup(self): # tracing at startup code = 'import tracemalloc; print(tracemalloc.is_tracing())' ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='1') stdout = stdout.rstrip() self.assertEqual(stdout, b'True') + def test_env_limit(self): # start and set the number of frames code = 'import tracemalloc; print(tracemalloc.get_traceback_limit())' ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='10') -- cgit v1.2.1