@@ -218,26 +218,22 @@ def test_rejects_empty_string(self):
218218 def test_rejects_multi_character_strings (self ):
219219 for s in ["***" , "AA" , "aA*!" ]:
220220 with self .subTest (echo_char = s ):
221- with self .assertRaises (ValueError ):
222- getpass .getpass (echo_char = s )
221+ self .assertRaises (ValueError , getpass .getpass , echo_char = s )
223222
224223 def test_rejects_non_ascii (self ):
225224 for ch in ["Æ" , "❤️" , "🐍" ]:
226225 with self .subTest (echo_char = ch ):
227- with self .assertRaises (ValueError ):
228- getpass .getpass (echo_char = ch )
226+ self .assertRaises (ValueError , getpass .getpass , echo_char = ch )
229227
230228 def test_rejects_control_characters (self ):
231229 for ch in ["\n " , "\t " , "\r " , "\x00 " , "\x7f " , "\x07 " ]:
232230 with self .subTest (echo_char = ch ):
233- with self .assertRaises (ValueError ):
234- getpass .getpass (echo_char = ch )
231+ self .assertRaises (ValueError , getpass .getpass , echo_char = ch )
235232
236233 def test_rejects_non_string (self ):
237234 for item in [b"*" , 0 , 0.0 , [], {}]:
238235 with self .subTest (echo_char = item ):
239- with self .assertRaises (TypeError ):
240- getpass .getpass (echo_char = item )
236+ self .assertRaises (TypeError , getpass .getpass , echo_char = item )
241237
242238
243239if __name__ == "__main__" :
0 commit comments