forked from huggingface/optimum-habana
-
Notifications
You must be signed in to change notification settings - Fork 21
/
conftest.py
25 lines (18 loc) · 679 Bytes
/
conftest.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class Secret:
"""
Taken from: https://stackoverflow.com/a/67393351
"""
def __init__(self, value):
self.value = value
def __repr__(self):
return "Secret(********)"
def __str___(self):
return "*******"
def pytest_addoption(parser):
parser.addoption("--token", action="store", default=None)
def pytest_generate_tests(metafunc):
# This is called for every test. Only get/set command line arguments
# if the argument is specified in the list of test "fixturenames".
option_value = Secret(metafunc.config.option.token)
if "token" in metafunc.fixturenames:
metafunc.parametrize("token", [option_value])