Reinforce testenv in ax test case (facebook#2530)

Summary: Pull Request resolved: facebook#2530

Differential Revision: D58842047

ax/utils/common/constants.py

@@ -87,3 +87,6 @@ class Keys(str, Enum):
 
 DEFAULT_WINSORIZATION_LIMITS_MINIMIZATION: Tuple[float, float] = (0.0, 0.2)
 DEFAULT_WINSORIZATION_LIMITS_MAXIMIZATION: Tuple[float, float] = (0.2, 0.0)
+
+TESTENV_ENV_KEY = "TESTENV"
+TESTENV_ENV_VAL = "True"

ax/utils/common/testutils.py

@@ -14,6 +14,7 @@
 import io
 import linecache
 import logging
+import os
 import signal
 import sys
 import types
@@ -39,6 +40,7 @@
 import numpy as np
 from ax.exceptions.core import AxParameterWarning
 from ax.utils.common.base import Base
+from ax.utils.common.constants import TESTENV_ENV_KEY, TESTENV_ENV_VAL
 from ax.utils.common.equality import object_attribute_dicts_find_unequal_fields
 from ax.utils.common.logger import get_logger
 from botorch.exceptions.warnings import InputDataWarning
@@ -301,6 +303,7 @@ def signal_handler(signum: int, frame: Optional[FrameType]) -> None:
 
         super().__init__(methodName=methodName)
         signal.signal(signal.SIGALRM, signal_handler)
+        os.environ[TESTENV_ENV_KEY] = TESTENV_ENV_VAL
 
     def setUp(self) -> None:
         """