File tree Expand file tree Collapse file tree 2 files changed +11
-7
lines changed Expand file tree Collapse file tree 2 files changed +11
-7
lines changed Original file line number Diff line number Diff line change @@ -599,12 +599,11 @@ def _from_subplot_args(figure, args):
599
599
)
600
600
i , j = num
601
601
else :
602
- if not isinstance (num , Integral ):
602
+ if not isinstance (num , Integral ) or num < 1 or num > rows * cols :
603
603
raise ValueError (
604
- f"Subplot specifier must be an integer, not { num !r} " )
605
- if num < 1 or num > rows * cols :
606
- raise ValueError (
607
- f"num must be 1 <= num <= { rows * cols } , not { num !r} " )
604
+ f"num must be an integer with 1 <= num <= { rows * cols } , "
605
+ f"not { num !r} "
606
+ )
608
607
i = j = num
609
608
return gs [i - 1 :j ]
610
609
Original file line number Diff line number Diff line change @@ -236,10 +236,15 @@ def test_add_subplot_invalid():
236
236
with pytest .raises (ValueError ,
237
237
match = 'Number of rows must be a positive integer' ):
238
238
fig .add_subplot (0 , 2 , 1 )
239
- with pytest .raises (ValueError , match = 'num must be 1 <= num <= 4' ):
239
+ with pytest .raises (ValueError , match = 'num must be an integer with '
240
+ '1 <= num <= 4' ):
240
241
fig .add_subplot (2 , 2 , 0 )
241
- with pytest .raises (ValueError , match = 'num must be 1 <= num <= 4' ):
242
+ with pytest .raises (ValueError , match = 'num must be an integer with '
243
+ '1 <= num <= 4' ):
242
244
fig .add_subplot (2 , 2 , 5 )
245
+ with pytest .raises (ValueError , match = 'num must be an integer with '
246
+ '1 <= num <= 4' ):
247
+ fig .add_subplot (2 , 2 , 0.5 )
243
248
244
249
with pytest .raises (ValueError , match = 'must be a three-digit integer' ):
245
250
fig .add_subplot (42 )
You can’t perform that action at this time.
0 commit comments