JezK
Edit File: ufunc_config.py
"""Typing tests for `numpy.core._ufunc_config`.""" import numpy as np def func1(a: str, b: int) -> None: ... def func2(a: str, b: int, c: float = ...) -> None: ... def func3(a: str, b: int) -> int: ... class Write1: def write(self, a: str) -> None: ... class Write2: def write(self, a: str, b: int = ...) -> None: ... class Write3: def write(self, a: str) -> int: ... _err_default = np.geterr() _bufsize_default = np.getbufsize() _errcall_default = np.geterrcall() try: np.seterr(all=None) np.seterr(divide="ignore") np.seterr(over="warn") np.seterr(under="call") np.seterr(invalid="raise") np.geterr() np.setbufsize(4096) np.getbufsize() np.seterrcall(func1) np.seterrcall(func2) np.seterrcall(func3) np.seterrcall(Write1()) np.seterrcall(Write2()) np.seterrcall(Write3()) np.geterrcall() with np.errstate(call=func1, all="call"): pass with np.errstate(call=Write1(), divide="log", over="log"): pass finally: np.seterr(**_err_default) np.setbufsize(_bufsize_default) np.seterrcall(_errcall_default)