@@ -39,16 +39,11 @@ def pytest_addoption(parser):
39
39
help = "keep temporary test directories" ,
40
40
)
41
41
parser .addoption (
42
- "--new-resolver" ,
43
- action = "store_true" ,
44
- default = False ,
45
- help = "use new resolver in tests" ,
46
- )
47
- parser .addoption (
48
- "--new-resolver-runtests" ,
49
- action = "store_true" ,
50
- default = False ,
51
- help = "run the skipped tests for the new resolver" ,
42
+ "--resolver" ,
43
+ action = "store" ,
44
+ default = "2020-resolver" ,
45
+ choices = ["2020-resolver" , "legacy" ],
46
+ help = "use given resolver in tests" ,
52
47
)
53
48
parser .addoption (
54
49
"--use-venv" ,
@@ -100,17 +95,17 @@ def pytest_collection_modifyitems(config, items):
100
95
def resolver_variant (request ):
101
96
"""Set environment variable to make pip default to the correct resolver.
102
97
"""
103
- new_resolver = request .config .getoption ("--new-resolver" )
98
+ resolver = request .config .getoption ("--resolver" )
99
+
100
+ # Handle the environment variables for this test.
104
101
features = set (os .environ .get ("PIP_USE_FEATURE" , "" ).split ())
105
102
if new_resolver :
106
- retval = "2020-resolver"
107
103
features .add ("2020-resolver" )
108
104
else :
109
- retval = "legacy"
110
105
features .discard ("2020-resolver" )
111
106
112
107
with patch .dict (os .environ , {"PIP_USE_FEATURE" : " " .join (features )}):
113
- yield retval
108
+ yield resolver
114
109
115
110
116
111
@pytest .fixture (scope = 'session' )
0 commit comments