7
7
from distutils .version import LooseVersion
8
8
import io
9
9
10
- from nose .tools import assert_raises
11
- from nose .plugins .skip import SkipTest
12
-
13
10
import datetime
14
11
15
12
import pytz
@@ -375,7 +372,8 @@ def test_shaped_data():
375
372
plt .plot (y2 )
376
373
377
374
plt .subplot (413 )
378
- assert_raises (ValueError , plt .plot , (y1 , y2 ))
375
+ with pytest .raises (ValueError ):
376
+ plt .plot ((y1 , y2 ))
379
377
380
378
plt .subplot (414 )
381
379
plt .plot (xdata [:, 1 ], xdata [1 , :], 'o' )
@@ -863,12 +861,14 @@ def test_pcolorargs():
863
861
Z = np .sqrt (X ** 2 + Y ** 2 )/ 5
864
862
865
863
_ , ax = plt .subplots ()
866
- assert_raises (TypeError , ax .pcolormesh , y , x , Z )
867
- assert_raises (TypeError , ax .pcolormesh , X , Y , Z .T )
868
- assert_raises (TypeError , ax .pcolormesh , x , y , Z [:- 1 , :- 1 ],
869
- shading = "gouraud" )
870
- assert_raises (TypeError , ax .pcolormesh , X , Y , Z [:- 1 , :- 1 ],
871
- shading = "gouraud" )
864
+ with pytest .raises (TypeError ):
865
+ ax .pcolormesh (y , x , Z )
866
+ with pytest .raises (TypeError ):
867
+ ax .pcolormesh (X , Y , Z .T )
868
+ with pytest .raises (TypeError ):
869
+ ax .pcolormesh (x , y , Z [:- 1 , :- 1 ], shading = "gouraud" )
870
+ with pytest .raises (TypeError ):
871
+ ax .pcolormesh (X , Y , Z [:- 1 , :- 1 ], shading = "gouraud" )
872
872
873
873
874
874
@image_comparison (baseline_images = ['canonical' ])
@@ -1316,13 +1316,15 @@ def test_scatter_2D():
1316
1316
fig , ax = plt .subplots ()
1317
1317
ax .scatter (x , y , c = z , s = 200 , edgecolors = 'face' )
1318
1318
1319
+
1319
1320
@cleanup
1320
1321
def test_scatter_color ():
1321
1322
# Try to catch cases where 'c' kwarg should have been used.
1322
- assert_raises (ValueError , plt .scatter , [1 , 2 ], [1 , 2 ],
1323
- color = [0.1 , 0.2 ])
1324
- assert_raises (ValueError , plt .scatter , [1 , 2 , 3 ], [1 , 2 , 3 ],
1325
- color = [1 , 2 , 3 ])
1323
+ with pytest .raises (ValueError ):
1324
+ plt .scatter ([1 , 2 ], [1 , 2 ], color = [0.1 , 0.2 ])
1325
+ with pytest .raises (ValueError ):
1326
+ plt .scatter ([1 , 2 , 3 ], [1 , 2 , 3 ], color = [1 , 2 , 3 ])
1327
+
1326
1328
1327
1329
@cleanup
1328
1330
def test_as_mpl_axes_api ():
@@ -1813,7 +1815,8 @@ def test_bxp_bad_widths():
1813
1815
1814
1816
fig , ax = plt .subplots ()
1815
1817
ax .set_yscale ('log' )
1816
- assert_raises (ValueError , ax .bxp , logstats , widths = [1 ])
1818
+ with pytest .raises (ValueError ):
1819
+ ax .bxp (logstats , widths = [1 ])
1817
1820
1818
1821
1819
1822
@cleanup
@@ -1825,7 +1828,8 @@ def test_bxp_bad_positions():
1825
1828
1826
1829
fig , ax = plt .subplots ()
1827
1830
ax .set_yscale ('log' )
1828
- assert_raises (ValueError , ax .bxp , logstats , positions = [2 , 3 ])
1831
+ with pytest .raises (ValueError ):
1832
+ ax .bxp (logstats , positions = [2 , 3 ])
1829
1833
1830
1834
1831
1835
@image_comparison (baseline_images = ['boxplot' , 'boxplot' ],
@@ -2010,24 +2014,26 @@ def test_boxplot_bad_medians_1():
2010
2014
x = np .linspace (- 7 , 7 , 140 )
2011
2015
x = np .hstack ([- 25 , x , 25 ])
2012
2016
fig , ax = plt .subplots ()
2013
- assert_raises (ValueError , ax .boxplot , x , usermedians = [1 , 2 ])
2017
+ with pytest .raises (ValueError ):
2018
+ ax .boxplot (x , usermedians = [1 , 2 ])
2014
2019
2015
2020
2016
2021
@cleanup
2017
2022
def test_boxplot_bad_medians_2 ():
2018
2023
x = np .linspace (- 7 , 7 , 140 )
2019
2024
x = np .hstack ([- 25 , x , 25 ])
2020
2025
fig , ax = plt .subplots ()
2021
- assert_raises (ValueError , ax .boxplot , [x , x ], usermedians = [[1 , 2 ], [1 , 2 ]])
2026
+ with pytest .raises (ValueError ):
2027
+ ax .boxplot ([x , x ], usermedians = [[1 , 2 ], [1 , 2 ]])
2022
2028
2023
2029
2024
2030
@cleanup
2025
2031
def test_boxplot_bad_ci_1 ():
2026
2032
x = np .linspace (- 7 , 7 , 140 )
2027
2033
x = np .hstack ([- 25 , x , 25 ])
2028
2034
fig , ax = plt .subplots ()
2029
- assert_raises ( ValueError , ax . boxplot , [ x , x ],
2030
- conf_intervals = [[1 , 2 ]])
2035
+ with pytest . raises ( ValueError ):
2036
+ ax . boxplot ([ x , x ], conf_intervals = [[1 , 2 ]])
2031
2037
2032
2038
2033
2039
@cleanup
@@ -2043,8 +2049,8 @@ def test_boxplot_bad_ci_2():
2043
2049
x = np .linspace (- 7 , 7 , 140 )
2044
2050
x = np .hstack ([- 25 , x , 25 ])
2045
2051
fig , ax = plt .subplots ()
2046
- assert_raises ( ValueError , ax . boxplot , [ x , x ],
2047
- conf_intervals = [[1 , 2 ], [1 ]])
2052
+ with pytest . raises ( ValueError ):
2053
+ ax . boxplot ([ x , x ], conf_intervals = [[1 , 2 ], [1 ]])
2048
2054
2049
2055
2050
2056
@image_comparison (baseline_images = ['boxplot_mod_artists_after_plotting' ],
@@ -2227,7 +2233,8 @@ def test_violinplot_bad_positions():
2227
2233
# First 9 digits of frac(sqrt(47))
2228
2234
np .random .seed (855654600 )
2229
2235
data = [np .random .normal (size = 100 ) for i in range (4 )]
2230
- assert_raises (ValueError , ax .violinplot , data , positions = range (5 ))
2236
+ with pytest .raises (ValueError ):
2237
+ ax .violinplot (data , positions = range (5 ))
2231
2238
2232
2239
2233
2240
@cleanup
@@ -2236,8 +2243,8 @@ def test_violinplot_bad_widths():
2236
2243
# First 9 digits of frac(sqrt(53))
2237
2244
np .random .seed (280109889 )
2238
2245
data = [np .random .normal (size = 100 ) for i in range (4 )]
2239
- assert_raises ( ValueError , ax . violinplot , data , positions = range ( 4 ),
2240
- widths = [1 , 2 , 3 ])
2246
+ with pytest . raises ( ValueError ):
2247
+ ax . violinplot ( data , positions = range ( 4 ), widths = [1 , 2 , 3 ])
2241
2248
2242
2249
2243
2250
@cleanup
@@ -2319,9 +2326,12 @@ def test_errorbar_shape():
2319
2326
yerr = np .vstack ((yerr1 , 2 * yerr1 )).T
2320
2327
xerr = 0.1 + yerr
2321
2328
2322
- assert_raises (ValueError , ax .errorbar , x , y , yerr = yerr , fmt = 'o' )
2323
- assert_raises (ValueError , ax .errorbar , x , y , xerr = xerr , fmt = 'o' )
2324
- assert_raises (ValueError , ax .errorbar , x , y , yerr = yerr , xerr = xerr , fmt = 'o' )
2329
+ with pytest .raises (ValueError ):
2330
+ ax .errorbar (x , y , yerr = yerr , fmt = 'o' )
2331
+ with pytest .raises (ValueError ):
2332
+ ax .errorbar (x , y , xerr = xerr , fmt = 'o' )
2333
+ with pytest .raises (ValueError ):
2334
+ ax .errorbar (x , y , yerr = yerr , xerr = xerr , fmt = 'o' )
2325
2335
2326
2336
2327
2337
@image_comparison (baseline_images = ['errorbar_limits' ])
@@ -3119,17 +3129,20 @@ def test_specgram_angle_freqs():
3119
3129
spec13 = ax13 .specgram (y , NFFT = NFFT , Fs = Fs , noverlap = noverlap ,
3120
3130
pad_to = pad_to , sides = 'twosided' , mode = 'angle' )
3121
3131
3122
- assert_raises (ValueError , ax11 .specgram , y , NFFT = NFFT , Fs = Fs ,
3123
- noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3124
- mode = 'phase' , scale = 'dB' )
3132
+ with pytest .raises (ValueError ):
3133
+ ax11 .specgram (y , NFFT = NFFT , Fs = Fs ,
3134
+ noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3135
+ mode = 'phase' , scale = 'dB' )
3125
3136
3126
- assert_raises (ValueError , ax12 .specgram , y , NFFT = NFFT , Fs = Fs ,
3127
- noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3128
- mode = 'phase' , scale = 'dB' )
3137
+ with pytest .raises (ValueError ):
3138
+ ax12 .specgram (y , NFFT = NFFT , Fs = Fs ,
3139
+ noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3140
+ mode = 'phase' , scale = 'dB' )
3129
3141
3130
- assert_raises (ValueError , ax13 .specgram , y , NFFT = NFFT , Fs = Fs ,
3131
- noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3132
- mode = 'phase' , scale = 'dB' )
3142
+ with pytest .raises (ValueError ):
3143
+ ax13 .specgram (y , NFFT = NFFT , Fs = Fs ,
3144
+ noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3145
+ mode = 'phase' , scale = 'dB' )
3133
3146
3134
3147
3135
3148
@image_comparison (baseline_images = ['specgram_angle_noise' ],
@@ -3166,17 +3179,20 @@ def test_specgram_noise_angle():
3166
3179
spec13 = ax13 .specgram (y , NFFT = NFFT , Fs = Fs , noverlap = noverlap ,
3167
3180
pad_to = pad_to , sides = 'twosided' , mode = 'angle' )
3168
3181
3169
- assert_raises (ValueError , ax11 .specgram , y , NFFT = NFFT , Fs = Fs ,
3170
- noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3171
- mode = 'phase' , scale = 'dB' )
3182
+ with pytest .raises (ValueError ):
3183
+ ax11 .specgram (y , NFFT = NFFT , Fs = Fs ,
3184
+ noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3185
+ mode = 'phase' , scale = 'dB' )
3172
3186
3173
- assert_raises (ValueError , ax12 .specgram , y , NFFT = NFFT , Fs = Fs ,
3174
- noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3175
- mode = 'phase' , scale = 'dB' )
3187
+ with pytest .raises (ValueError ):
3188
+ ax12 .specgram (y , NFFT = NFFT , Fs = Fs ,
3189
+ noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3190
+ mode = 'phase' , scale = 'dB' )
3176
3191
3177
- assert_raises (ValueError , ax13 .specgram , y , NFFT = NFFT , Fs = Fs ,
3178
- noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3179
- mode = 'phase' , scale = 'dB' )
3192
+ with pytest .raises (ValueError ):
3193
+ ax13 .specgram (y , NFFT = NFFT , Fs = Fs ,
3194
+ noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3195
+ mode = 'phase' , scale = 'dB' )
3180
3196
3181
3197
3182
3198
@image_comparison (baseline_images = ['specgram_phase_freqs' ],
@@ -3221,17 +3237,20 @@ def test_specgram_freqs_phase():
3221
3237
spec13 = ax13 .specgram (y , NFFT = NFFT , Fs = Fs , noverlap = noverlap ,
3222
3238
pad_to = pad_to , sides = 'twosided' , mode = 'phase' )
3223
3239
3224
- assert_raises (ValueError , ax11 .specgram , y , NFFT = NFFT , Fs = Fs ,
3225
- noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3226
- mode = 'phase' , scale = 'dB' )
3240
+ with pytest .raises (ValueError ):
3241
+ ax11 .specgram (y , NFFT = NFFT , Fs = Fs ,
3242
+ noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3243
+ mode = 'phase' , scale = 'dB' )
3227
3244
3228
- assert_raises (ValueError , ax12 .specgram , y , NFFT = NFFT , Fs = Fs ,
3229
- noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3230
- mode = 'phase' , scale = 'dB' )
3245
+ with pytest .raises (ValueError ):
3246
+ ax12 .specgram (y , NFFT = NFFT , Fs = Fs ,
3247
+ noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3248
+ mode = 'phase' , scale = 'dB' )
3231
3249
3232
- assert_raises (ValueError , ax13 .specgram , y , NFFT = NFFT , Fs = Fs ,
3233
- noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3234
- mode = 'phase' , scale = 'dB' )
3250
+ with pytest .raises (ValueError ):
3251
+ ax13 .specgram (y , NFFT = NFFT , Fs = Fs ,
3252
+ noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3253
+ mode = 'phase' , scale = 'dB' )
3235
3254
3236
3255
3237
3256
@image_comparison (baseline_images = ['specgram_phase_noise' ],
@@ -3271,17 +3290,20 @@ def test_specgram_noise_phase():
3271
3290
pad_to = pad_to , sides = 'twosided' ,
3272
3291
mode = 'phase' , )
3273
3292
3274
- assert_raises (ValueError , ax11 .specgram , y , NFFT = NFFT , Fs = Fs ,
3275
- noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3276
- mode = 'phase' , scale = 'dB' )
3293
+ with pytest .raises (ValueError ):
3294
+ ax11 .specgram (y , NFFT = NFFT , Fs = Fs ,
3295
+ noverlap = noverlap , pad_to = pad_to , sides = 'default' ,
3296
+ mode = 'phase' , scale = 'dB' )
3277
3297
3278
- assert_raises (ValueError , ax12 .specgram , y , NFFT = NFFT , Fs = Fs ,
3279
- noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3280
- mode = 'phase' , scale = 'dB' )
3298
+ with pytest .raises (ValueError ):
3299
+ ax12 .specgram (y , NFFT = NFFT , Fs = Fs ,
3300
+ noverlap = noverlap , pad_to = pad_to , sides = 'onesided' ,
3301
+ mode = 'phase' , scale = 'dB' )
3281
3302
3282
- assert_raises (ValueError , ax13 .specgram , y , NFFT = NFFT , Fs = Fs ,
3283
- noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3284
- mode = 'phase' , scale = 'dB' )
3303
+ with pytest .raises (ValueError ):
3304
+ ax13 .specgram (y , NFFT = NFFT , Fs = Fs ,
3305
+ noverlap = noverlap , pad_to = pad_to , sides = 'twosided' ,
3306
+ mode = 'phase' , scale = 'dB' )
3285
3307
3286
3308
3287
3309
@image_comparison (baseline_images = ['psd_freqs' ], remove_text = True ,
@@ -4230,8 +4252,10 @@ def test_square_plot():
4230
4252
@cleanup
4231
4253
def test_no_None ():
4232
4254
fig , ax = plt .subplots ()
4233
- assert_raises (ValueError , plt .plot , None )
4234
- assert_raises (ValueError , plt .plot , None , None )
4255
+ with pytest .raises (ValueError ):
4256
+ plt .plot (None )
4257
+ with pytest .raises (ValueError ):
4258
+ plt .plot (None , None )
4235
4259
4236
4260
4237
4261
@cleanup
@@ -4335,8 +4359,10 @@ def test_title_location_roundtrip():
4335
4359
assert 'right' == ax .get_title (loc = 'right' )
4336
4360
assert 'aardvark' == ax .get_title ()
4337
4361
4338
- assert_raises (ValueError , ax .get_title , loc = 'foo' )
4339
- assert_raises (ValueError , ax .set_title , 'fail' , loc = 'foo' )
4362
+ with pytest .raises (ValueError ):
4363
+ ax .get_title (loc = 'foo' )
4364
+ with pytest .raises (ValueError ):
4365
+ ax .set_title ('fail' , loc = 'foo' )
4340
4366
4341
4367
4342
4368
@image_comparison (baseline_images = ["loglog" ], remove_text = True ,
@@ -4484,7 +4510,7 @@ def test_pandas_indexing_dates():
4484
4510
try :
4485
4511
import pandas as pd
4486
4512
except ImportError :
4487
- raise SkipTest ("Pandas not installed" )
4513
+ pytest . skip ("Pandas not installed" )
4488
4514
4489
4515
dates = np .arange ('2005-02' , '2005-03' , dtype = 'datetime64[D]' )
4490
4516
values = np .sin (np .array (range (len (dates ))))
@@ -4501,7 +4527,7 @@ def test_pandas_errorbar_indexing():
4501
4527
try :
4502
4528
import pandas as pd
4503
4529
except ImportError :
4504
- raise SkipTest ("Pandas not installed" )
4530
+ pytest . skip ("Pandas not installed" )
4505
4531
4506
4532
df = pd .DataFrame (np .random .uniform (size = (5 , 4 )),
4507
4533
columns = ['x' , 'y' , 'xe' , 'ye' ],
@@ -4515,7 +4541,7 @@ def test_pandas_indexing_hist():
4515
4541
try :
4516
4542
import pandas as pd
4517
4543
except ImportError :
4518
- raise SkipTest ("Pandas not installed" )
4544
+ pytest . skip ("Pandas not installed" )
4519
4545
4520
4546
ser_1 = pd .Series (data = [1 , 2 , 2 , 3 , 3 , 4 , 4 , 4 , 4 , 5 ])
4521
4547
ser_2 = ser_1 .iloc [1 :]
@@ -4545,8 +4571,8 @@ def test_none_kwargs():
4545
4571
4546
4572
@cleanup
4547
4573
def test_ls_ds_conflict ():
4548
- assert_raises ( ValueError , plt . plot , range ( 32 ),
4549
- linestyle = 'steps-pre:' , drawstyle = 'steps-post' )
4574
+ with pytest . raises ( ValueError ):
4575
+ plt . plot ( range ( 32 ), linestyle = 'steps-pre:' , drawstyle = 'steps-post' )
4550
4576
4551
4577
4552
4578
@image_comparison (baseline_images = ['date_timezone_x' ], extensions = ['png' ])
0 commit comments