@@ -181,18 +181,25 @@ def test_set_quiet(base_app):
181
181
out , err = run_cmd (base_app , 'set quiet' )
182
182
assert out == ['quiet: True' ]
183
183
184
- @pytest .mark .parametrize ('new_val, is_valid' , [
185
- (ansi .ANSI_NEVER , False ),
186
- (ansi . ANSI_TERMINAL , False ),
187
- (ansi .ANSI_ALWAYS , False ),
188
- ('neVeR ' , False ),
189
- ('TeRMInal' , False ),
190
- ('AlWaYs' , False ),
191
- ('invalid' , True ),
184
+ @pytest .mark .parametrize ('new_val, is_valid, expected ' , [
185
+ (ansi .ANSI_NEVER , False , ansi . ANSI_NEVER ),
186
+ ('neVeR' , False , ansi . ANSI_NEVER ),
187
+ (ansi .ANSI_TERMINAL , False , ansi . ANSI_TERMINAL ),
188
+ ('TeRMInal ' , False , ansi . ANSI_TERMINAL ),
189
+ (ansi . ANSI_ALWAYS , False , ansi . ANSI_ALWAYS ),
190
+ ('AlWaYs' , False , ansi . ANSI_ALWAYS ),
191
+ ('invalid' , True , ansi . ANSI_TERMINAL ),
192
192
])
193
- def test_set_allow_ansi (base_app , new_val , is_valid ):
193
+ def test_set_allow_ansi (base_app , new_val , is_valid , expected ):
194
+ # Initialize allow_ansi for this test
195
+ ansi .allow_ansi = ansi .ANSI_TERMINAL
196
+
197
+ # Use the set command to alter it
194
198
out , err = run_cmd (base_app , 'set allow_ansi {}' .format (new_val ))
199
+
200
+ # Verify the results
195
201
assert bool (err ) == is_valid
202
+ assert ansi .allow_ansi == expected
196
203
197
204
# Reload ansi module to reset allow_ansi to its default since it's an
198
205
# application-wide setting that can affect other unit tests.
0 commit comments