@@ -196,10 +196,13 @@ def test_ampm(self):
196196 pm_am = time .strftime ("%p" );
197197 if pm_am == 'AM' or pm_am == 'am' or pm_am == 'PM' or pm_am == 'pm' :
198198 # the test has sence only if the pm/am is provided
199- self .check_format ("%p" , (2018 , 2 , 18 , 0 , 0 , 0 , 3 , 1 , 0 ), 'AM' )
200- self .check_format ("%p" , (2018 , 8 , 18 , 11 , 8 , 0 , 3 , 1 , 0 ), 'AM' )
201- self .check_format ("%p" , (2018 , 8 , 18 , 12 , 50 , 1 , 3 , 1 , 0 ), 'PM' )
202- self .check_format ("%p" , (2018 , 8 , 18 , 23 , 59 , 0 , 3 , 1 , 0 ), 'PM' )
199+ is_lower_case = pm_am [1 ] == 'm'
200+ pm_case = 'pm' if is_lower_case else 'PM'
201+ am_case = 'am' if is_lower_case else 'AM'
202+ self .check_format ("%p" , (2018 , 2 , 18 , 0 , 0 , 0 , 3 , 1 , 0 ), am_case )
203+ self .check_format ("%p" , (2018 , 8 , 18 , 11 , 8 , 0 , 3 , 1 , 0 ), am_case )
204+ self .check_format ("%p" , (2018 , 8 , 18 , 12 , 50 , 1 , 3 , 1 , 0 ), pm_case )
205+ self .check_format ("%p" , (2018 , 8 , 18 , 23 , 59 , 0 , 3 , 1 , 0 ), pm_case )
203206 self .assertRaises (ValueError , time .strftime , "%p" , time .struct_time ((2018 , 8 , 2 , - 1 , 24 , 30 , 1 , 1 , 0 )))
204207 self .assertRaises (ValueError , time .strftime , "%p" , time .struct_time ((2018 , 8 , 2 , 24 , 24 , 30 , 1 , 1 , 0 )))
205208
0 commit comments