@@ -223,26 +223,33 @@ def test_guess_known_extensions(self):
223
223
224
224
def test_preferred_extension (self ):
225
225
def check_extensions ():
226
- self .assertEqual (mimetypes .guess_extension ('application/octet-stream' ), '.bin' )
227
- self .assertEqual (mimetypes .guess_extension ('application/postscript' ), '.ps' )
228
- self .assertEqual (mimetypes .guess_extension ('application/vnd.apple.mpegurl' ), '.m3u' )
229
- self .assertEqual (mimetypes .guess_extension ('application/vnd.ms-excel' ), '.xls' )
230
- self .assertEqual (mimetypes .guess_extension ('application/vnd.ms-powerpoint' ), '.ppt' )
231
- self .assertEqual (mimetypes .guess_extension ('application/x-texinfo' ), '.texi' )
232
- self .assertEqual (mimetypes .guess_extension ('application/x-troff' ), '.roff' )
233
- self .assertEqual (mimetypes .guess_extension ('application/xml' ), '.xsl' )
234
- self .assertEqual (mimetypes .guess_extension ('audio/mpeg' ), '.mp3' )
235
- self .assertEqual (mimetypes .guess_extension ('image/avif' ), '.avif' )
236
- self .assertEqual (mimetypes .guess_extension ('image/webp' ), '.webp' )
237
- self .assertEqual (mimetypes .guess_extension ('image/jpeg' ), '.jpg' )
238
- self .assertEqual (mimetypes .guess_extension ('image/tiff' ), '.tiff' )
239
- self .assertEqual (mimetypes .guess_extension ('message/rfc822' ), '.eml' )
240
- self .assertEqual (mimetypes .guess_extension ('text/html' ), '.html' )
241
- self .assertEqual (mimetypes .guess_extension ('text/plain' ), '.txt' )
242
- self .assertEqual (mimetypes .guess_extension ('text/rtf' ), '.rtf' )
243
- self .assertEqual (mimetypes .guess_extension ('text/x-rst' ), '.rst' )
244
- self .assertEqual (mimetypes .guess_extension ('video/mpeg' ), '.mpeg' )
245
- self .assertEqual (mimetypes .guess_extension ('video/quicktime' ), '.mov' )
226
+ for mime_type , ext in (
227
+ ("application/octet-stream" , ".bin" ),
228
+ ("application/postscript" , ".ps" ),
229
+ ("application/vnd.apple.mpegurl" , ".m3u" ),
230
+ ("application/vnd.ms-excel" , ".xls" ),
231
+ ("application/vnd.ms-powerpoint" , ".ppt" ),
232
+ ("application/x-texinfo" , ".texi" ),
233
+ ("application/x-troff" , ".roff" ),
234
+ ("application/xml" , ".xsl" ),
235
+ ("audio/matroska" , ".mka" ),
236
+ ("audio/mpeg" , ".mp3" ),
237
+ ("image/avif" , ".avif" ),
238
+ ("image/webp" , ".webp" ),
239
+ ("image/jpeg" , ".jpg" ),
240
+ ("image/tiff" , ".tiff" ),
241
+ ("message/rfc822" , ".eml" ),
242
+ ("text/html" , ".html" ),
243
+ ("text/plain" , ".txt" ),
244
+ ("text/rtf" , ".rtf" ),
245
+ ("text/x-rst" , ".rst" ),
246
+ ("video/matroska" , ".mkv" ),
247
+ ("video/matroska-3d" , ".mk3d" ),
248
+ ("video/mpeg" , ".mpeg" ),
249
+ ("video/quicktime" , ".mov" ),
250
+ ):
251
+ with self .subTest (mime_type = mime_type , ext = ext ):
252
+ self .assertEqual (mimetypes .guess_extension (mime_type ), ext )
246
253
247
254
check_extensions ()
248
255
mimetypes .init ()
0 commit comments