Ñò
[ÐKc @ sH d d k Z e i d d j o d d Un e d j o e ƒ n d S( iÿÿÿÿNi i s$
from __future__ import with_statement
from numpy.core import *
from numpy.random import rand, randint
from numpy.testing import *
class TestErrstate(TestCase):
def test_invalid(self):
with errstate(all='raise', under='ignore'):
a = -arange(3)
# This should work
with errstate(invalid='ignore'):
sqrt(a)
# While this should fail!
try:
sqrt(a)
except FloatingPointError:
pass
else:
self.fail()
def test_divide(self):
with errstate(all='raise', under='ignore'):
a = -arange(3)
# This should work
with errstate(divide='ignore'):
a / 0
# While this should fail!
try:
a / 0
except FloatingPointError:
pass
else:
self.fail()
def test_errcall(self):
def foo(*args):
print args
olderrcall = geterrcall()
with errstate(call=foo):
assert(geterrcall() is foo), 'call is not foo'
with errstate(call=None):
assert(geterrcall() is None), 'call is not None'
assert(geterrcall() is olderrcall), 'call is not olderrcall'
t __main__( i i ( t syst version_infot __name__t run_module_suite( ( ( sD /usr/lib64/python2.6/site-packages/numpy/core/tests/test_errstate.pyt s .