1 files changed, 40 insertions, 0 deletions
diff --git a/numpy/numarray/util.py b/numpy/numarray/util.py
new file mode 100644
index 000000000..2a7efb60d
--- /dev/null
+++ b/numpy/numarray/util.py
@@ -0,0 +1,40 @@
+from numpy import geterr
+
+__all__ = ['MathDomainError', 'UnderflowError', 'NumOverflowError', 'handleError',
+           'get_numarray_include_dirs']
+
+class MathDomainError(ArithmeticError): pass
+class UnderflowError(ArithmeticError): pass
+class NumOverflowError(OverflowError, ArithmeticError): pass
+
+def handleError(errorStatus, sourcemsg):
+    """Take error status and use error mode to handle it."""
+    modes = geterr()
+    if errorStatus & FPE_INVALID:
+        if modes['invalid'] == "warn":
+            print "Warning: Encountered invalid numeric result(s)", sourcemsg
+        if modes['invalid'] == "raise":
+            raise MathDomainError(sourcemsg)
+    if errorStatus & FPE_DIVIDEBYZERO:
+        if modes['dividebyzero'] == "warn":
+            print "Warning: Encountered divide by zero(s)", sourcemsg
+        if modes['dividebyzero'] == "raise":
+            raise ZeroDivisionError(sourcemsg)
+    if errorStatus & FPE_OVERFLOW:
+        if modes['overflow'] == "warn":
+            print "Warning: Encountered overflow(s)", sourcemsg
+        if modes['overflow'] == "raise":
+            raise NumOverflowError(sourcemsg)
+    if errorStatus & FPE_UNDERFLOW:
+        if modes['underflow'] == "warn":
+            print "Warning: Encountered underflow(s)", sourcemsg
+        if modes['underflow'] == "raise":
+            raise UnderflowError(sourcemsg)
+
+
+import os
+import numpy
+def get_numarray_include_dirs():
+    base = os.path.dirname(numpy.__file__)
+    newdirs = [os.path.join(base, 'numarray')]
+    return newdirs