Commit 34cd2ae6 authored by Gregory P. Smith's avatar Gregory P. Smith

Break up TestCommandLine.test_env_var into four distinct tests.

parent 7ffa2c5f
...@@ -748,26 +748,28 @@ class TestFilters(unittest.TestCase): ...@@ -748,26 +748,28 @@ class TestFilters(unittest.TestCase):
class TestCommandLine(unittest.TestCase): class TestCommandLine(unittest.TestCase):
def test_env_var(self): def test_env_var_disabled_by_default(self):
# not tracing by default # not tracing by default
code = 'import tracemalloc; print(tracemalloc.is_tracing())' code = 'import tracemalloc; print(tracemalloc.is_tracing())'
ok, stdout, stderr = assert_python_ok('-c', code) ok, stdout, stderr = assert_python_ok('-c', code)
stdout = stdout.rstrip() stdout = stdout.rstrip()
self.assertEqual(stdout, b'False') self.assertEqual(stdout, b'False')
# PYTHON* environment variables must be ignored when -E option is def test_env_var_ignored_with_E(self):
# present """PYTHON* environment variables must be ignored when -E is present."""
code = 'import tracemalloc; print(tracemalloc.is_tracing())' code = 'import tracemalloc; print(tracemalloc.is_tracing())'
ok, stdout, stderr = assert_python_ok('-E', '-c', code, PYTHONTRACEMALLOC='1') ok, stdout, stderr = assert_python_ok('-E', '-c', code, PYTHONTRACEMALLOC='1')
stdout = stdout.rstrip() stdout = stdout.rstrip()
self.assertEqual(stdout, b'False') self.assertEqual(stdout, b'False')
def test_env_var_enabled_at_startup(self):
# tracing at startup # tracing at startup
code = 'import tracemalloc; print(tracemalloc.is_tracing())' code = 'import tracemalloc; print(tracemalloc.is_tracing())'
ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='1') ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='1')
stdout = stdout.rstrip() stdout = stdout.rstrip()
self.assertEqual(stdout, b'True') self.assertEqual(stdout, b'True')
def test_env_limit(self):
# start and set the number of frames # start and set the number of frames
code = 'import tracemalloc; print(tracemalloc.get_traceback_limit())' code = 'import tracemalloc; print(tracemalloc.get_traceback_limit())'
ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='10') ok, stdout, stderr = assert_python_ok('-c', code, PYTHONTRACEMALLOC='10')
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment