1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
from ...tests.helper import catch_warnings
from ..exceptions import AstropyDeprecationWarning
from ...extern.six.moves import reload_module
def test_import_warning():
# this import *maybe* raises the warning too, but we force the matter below
from ..compat import argparse # pylint: disable=W0611
with catch_warnings() as w:
# this ensures the warning will be re-issues at reload. Without this,
# the warning is missing if you invoke astropy.test twice in the same
# python session
if hasattr(argparse, '__warningregistry__'):
argparse.__warningregistry__.clear()
reload_module(argparse)
assert len(w) == 1
assert w[0].category == AstropyDeprecationWarning
assert w[0].message.args[0] == ("astropy.utils.compat.argparse is now "
"deprecated - use the argparse module "
"directly instead")
|
