| import numpy as np |
|
|
| try: |
| import mpmath |
| except ImportError: |
| pass |
|
|
|
|
| def mpmath_wrightomega(x): |
| return mpmath.lambertw(mpmath.exp(x), mpmath.mpf('-0.5')) |
|
|
|
|
| def wrightomega_series_error(x): |
| series = x |
| desired = mpmath_wrightomega(x) |
| return abs(series - desired) / desired |
|
|
|
|
| def wrightomega_exp_error(x): |
| exponential_approx = mpmath.exp(x) |
| desired = mpmath_wrightomega(x) |
| return abs(exponential_approx - desired) / desired |
|
|
|
|
| def main(): |
| desired_error = 2 * np.finfo(float).eps |
| print('Series Error') |
| for x in [1e5, 1e10, 1e15, 1e20]: |
| with mpmath.workdps(100): |
| error = wrightomega_series_error(x) |
| print(x, error, error < desired_error) |
|
|
| print('Exp error') |
| for x in [-10, -25, -50, -100, -200, -400, -700, -740]: |
| with mpmath.workdps(100): |
| error = wrightomega_exp_error(x) |
| print(x, error, error < desired_error) |
|
|
|
|
| if __name__ == '__main__': |
| main() |
|
|
