numpy.errstate decorator to catch floating point warnings as error

catch_warning_numpy.py

#!/usr/bin/env python3

# use numpy.errstate() decorator or context manager instead of numpy.seterr() directly 
# as seterr() interferes with all other functions.
# That is, scripts or packages that call this module would be affected by np.seterr()

import math
import numpy as np

@np.errstate(divide="raise", invalid="raise")
def zero_div(A):
    try:
        A / 0
        raise RuntimeError(f"should have raised {A} / 0", type(A))
    except ArithmeticError as e:
        print(f"caught {e}   {A} / 0", type(A))

@np.errstate(over="raise")
def overflow(A):
    try:
        np.exp(A)
    except ArithmeticError as e:
        print(f"caught {e}  exp({A})", type(A))

zero_div(1)
zero_div(0)
zero_div(np.float64(1))
zero_div(np.float64(0))
zero_div(np.float32(1))
zero_div(np.float32(0))
zero_div(np.float16(1))
zero_div(np.float16(0))

overflow(100000.)
overflow(np.float64(100000.))