25e140652b
This patch changes how most of the plain math functions are implemented:
there are now two generic math wrapper functions that take a pointer to a
math function (like sin, cos) and perform the necessary conversion to and
from MicroPython types. This helps to reduce code size. The generic
functions can also check for math domain errors in a generic way, by
testing if the result is NaN or infinity combined with finite inputs.
The result is that, with this patch, all math functions now have full
domain error checking (even gamma and lgamma) and code size has decreased
for most ports. Code size changes in bytes for those with the math module
are:
unix x64: -432
unix nanbox: -792
stm32: -88
esp8266: +12
Tests are also added to check domain errors are handled correctly.
37 lines
932 B
Python
37 lines
932 B
Python
# Tests domain errors in special math functions
|
|
|
|
try:
|
|
import math
|
|
math.erf
|
|
except (ImportError, AttributeError):
|
|
print("SKIP")
|
|
raise SystemExit
|
|
|
|
inf = float('inf')
|
|
nan = float('nan')
|
|
|
|
# single argument functions
|
|
for name, f, args in (
|
|
('expm1', math.exp, ()),
|
|
('log2', math.log2, (-1, 0)),
|
|
('log10', math.log10, (-1, 0)),
|
|
('sinh', math.sinh, ()),
|
|
('cosh', math.cosh, ()),
|
|
('tanh', math.tanh, ()),
|
|
('asinh', math.asinh, ()),
|
|
('acosh', math.acosh, (-1, 0.9, 1)),
|
|
('atanh', math.atanh, (-1, 1)),
|
|
('erf', math.erf, ()),
|
|
('erfc', math.erfc, ()),
|
|
('gamma', math.gamma, (-2, -1, 0, 1)),
|
|
('lgamma', math.lgamma, (-2, -1, 0, 1)),
|
|
):
|
|
for x in args + (inf, nan):
|
|
try:
|
|
ans = f(x)
|
|
print('%.4f' % ans)
|
|
except ValueError:
|
|
print(name, 'ValueError')
|
|
except OverflowError:
|
|
print(name, 'OverflowError')
|