def test_seterr_geterr(self):
default = _ufunc_config.geterr()
assert default['fallback_mode'] == 'ignore'
old = _ufunc_config.seterr(fallback_mode='warn')
current = _ufunc_config.geterr()
assert old['fallback_mode'] == 'ignore'
assert current['fallback_mode'] == 'warn'
_ufunc_config.seterr(**old)
def test_errstate(self):
old = _ufunc_config.seterr(fallback_mode='print')
before = _ufunc_config.geterr()
with _ufunc_config.errstate(fallback_mode='raise'):
inside = _ufunc_config.geterr()
assert inside['fallback_mode'] == 'raise'
after = _ufunc_config.geterr()
assert before['fallback_mode'] == after['fallback_mode']
_ufunc_config.seterr(**old)
def setUp(self):
old_config = _ufunc_config.geterr()
yield
_ufunc_config.seterr(**old_config)
def setUp(self):
self.old_config = _ufunc_config.geterr()