diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2013-07-11 16:49:04 -0600 |
|---|---|---|
| committer | Charles Harris <charlesr.harris@gmail.com> | 2013-07-11 16:49:04 -0600 |
| commit | 2f1174dee44e901b7d028beb86f4a8ea324bd74f (patch) | |
| tree | 8f09dc2bd35e2631f5821fe2e998f6ea46e254b8 /numpy/lib/tests/test_function_base.py | |
| parent | 49a587cd786242b05fcfd22d5cda961d733b68d4 (diff) | |
| download | numpy-2f1174dee44e901b7d028beb86f4a8ea324bd74f.tar.gz | |
MAINT: Use np.errstate context manager.
Now that Python < 2.6 is no longer supported we can use the errstate
context manager in places where constructs like
```
old = seterr(invalid='ignore')
try:
blah
finally:
seterr(**old)
```
were used.
Diffstat (limited to 'numpy/lib/tests/test_function_base.py')
| -rw-r--r-- | numpy/lib/tests/test_function_base.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/numpy/lib/tests/test_function_base.py b/numpy/lib/tests/test_function_base.py index 45e248913..a23e406e3 100644 --- a/numpy/lib/tests/test_function_base.py +++ b/numpy/lib/tests/test_function_base.py @@ -1062,8 +1062,7 @@ class TestHistogramdd(TestCase): def test_inf_edges(self): """Test using +/-inf bin edges works. See #1788.""" - olderr = np.seterr(invalid='ignore') - try: + with np.errstate(invalid='ignore'): x = np.arange(6).reshape(3, 2) expected = np.array([[1, 0], [0, 1], [0, 1]]) h, e = np.histogramdd(x, bins=[3, [-np.inf, 2, 10]]) @@ -1072,8 +1071,6 @@ class TestHistogramdd(TestCase): assert_allclose(h, expected) h, e = np.histogramdd(x, bins=[3, [-np.inf, 3, np.inf]]) assert_allclose(h, expected) - finally: - np.seterr(**olderr) class TestUnique(TestCase): |