Coverage for mlos_bench/mlos_bench/tests/test_with_alt_tz.py: 86%

1# 

2# Copyright (c) Microsoft Corporation. 

3# Licensed under the MIT License. 

4# 

5"""Tests various other test scenarios with alternative default (un-named) TZ info.""" 

6 

7import os 

8import sys 

9from subprocess import run 

10from typing import Optional 

11 

12import pytest 

13 

14from mlos_bench.tests import ZONE_NAMES 

15 

16DIRNAME = os.path.dirname(__file__) 

17TZ_TEST_FILES = [ 

18 DIRNAME + "/environments/local/composite_local_env_test.py", 

19 DIRNAME + "/environments/local/local_env_telemetry_test.py", 

20 DIRNAME + "/storage/exp_load_test.py", 

21 DIRNAME + "/storage/trial_telemetry_test.py", 

22] 

23 

24 

25@pytest.mark.skipif(sys.platform == "win32", reason="TZ environment variable is a UNIXism") 

26@pytest.mark.parametrize(("tz_name"), ZONE_NAMES) 

27@pytest.mark.parametrize(("test_file"), TZ_TEST_FILES) 

28def test_trial_telemetry_alt_tz(tz_name: Optional[str], test_file: str) -> None: 

29 """Run the tests under alternative default (un-named) TZ info.""" 

30 env = os.environ.copy() 

31 if tz_name is None: 

32 env.pop("TZ", None) 

33 else: 

34 env["TZ"] = tz_name 

35 cmd = run( 

36 [sys.executable, "-m", "pytest", "-n0", test_file], 

37 env=env, 

38 capture_output=True, 

39 check=False, 

40 ) 

41 if cmd.returncode != 0: 

42 print(cmd.stdout.decode()) 

43 print(cmd.stderr.decode()) 

44 raise AssertionError( 

45 f"Test(s) failed: # TZ='{tz_name}' '{sys.executable}' -m pytest -n0 '{test_file}'" 

46 )