Test new and legacy envs for setting standard tb

Author: colin-nolan

tests/test_tutorial/test_exceptions/test_tutorial001.py

@@ -3,6 +3,7 @@
 import sys
 from pathlib import Path
 
+import pytest
 from typer.testing import CliRunner
 
 from docs_src.exceptions import tutorial001 as mod
@@ -30,16 +31,18 @@ def test_traceback_rich():
     assert "name = 'morty'" in result.stderr
 
 
-def test_standard_traceback_env_var():
+@pytest.mark.parametrize(
+    "env_var", ["TYPER_STANDARD_TRACEBACK", "_TYPER_STANDARD_TRACEBACK"]
+)
+def test_standard_traceback_env_var(env_var: str):
     file_path = Path(mod.__file__)
     result = subprocess.run(
         [sys.executable, "-m", "coverage", "run", str(file_path)],
         capture_output=True,
         encoding="utf-8",
-        env={**os.environ, "TYPER_STANDARD_TRACEBACK": "1"},
+        env={**os.environ, env_var: "1"},
     )
     assert "return get_command(self)(*args, **kwargs)" in result.stderr
-
     assert "typer.run(main)" in result.stderr
     assert "print(name + 3)" in result.stderr
     assert 'TypeError: can only concatenate str (not "int") to str' in result.stderr

tests/test_tutorial/test_exceptions/test_tutorial002.py

@@ -3,6 +3,7 @@
 import sys
 from pathlib import Path
 
+import pytest
 from typer.testing import CliRunner
 
 from docs_src.exceptions import tutorial002 as mod
@@ -30,13 +31,16 @@ def test_traceback_rich():
     assert "name = 'morty'" not in result.stderr
 
 
-def test_standard_traceback_env_var():
+@pytest.mark.parametrize(
+    "env_var", ["TYPER_STANDARD_TRACEBACK", "_TYPER_STANDARD_TRACEBACK"]
+)
+def test_standard_traceback_env_var(env_var: str):
     file_path = Path(mod.__file__)
     result = subprocess.run(
         [sys.executable, "-m", "coverage", "run", str(file_path), "secret"],
         capture_output=True,
         encoding="utf-8",
-        env={**os.environ, "TYPER_STANDARD_TRACEBACK": "1"},
+        env={**os.environ, env_var: "1"},
     )
     assert "return get_command(self)(*args, **kwargs)" in result.stderr