@@ -181,18 +181,25 @@ def test_set_quiet(base_app):
181181 out , err = run_cmd (base_app , 'set quiet' )
182182 assert out == ['quiet: True' ]
183183
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 ),
192192])
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
194198 out , err = run_cmd (base_app , 'set allow_ansi {}' .format (new_val ))
199+
200+ # Verify the results
195201 assert bool (err ) == is_valid
202+ assert ansi .allow_ansi == expected
196203
197204 # Reload ansi module to reset allow_ansi to its default since it's an
198205 # application-wide setting that can affect other unit tests.
0 commit comments