numpy-flint-arb 1.3.0

Arbitrary precision ball arithmetic (interval arithmetic) dtype in NumPy

numpy-flint-arb

---

Documentation: https://numpy-flint-arb.readthedocs.io

Source Code: https://github.com/34j/numpy-flint-arb

---

Arbitrary precision ball arithmetic (interval arithmetic) dtype in NumPy

Installation

Install this via pip (or your favourite package manager):

pip install numpy-flint-arb

Usage

Import numpy_flint_arb.np instead of numpy:

from numpy_flint_arb import np

A = np.random.normal(size=(2, 2))
b = np.random.normal(size=(2,))
x = np.linalg.solve(A, b)
b_approx = A @ x
assert np.all(np.contains(b_approx, b))

Input Check

To avoid mixing ordinary floats like float or np.float, flarray for arb, acb only accepts integers, arb or acb and flarray for arf only accepts integers and arf, arb.

To relax this, allow_input() may be used:

import pytest
from numpy_flint_arb import allow_input
from flint import arb, arf

# arb array
with pytest.raises(Exception):
    np.asarray(0.5, dtype=arb)
with pytest.raises(Exception):
    with allow_input(float=True):
        np.asarray(0.5, dtype=arb)
with allow_input(interval=True, float=True):
    np.asarray(0.5, dtype=arb)

# arf array
with pytest.raises(Exception):
    np.asarray(0.5, dtype=arf)
with allow_input(float=True):
    np.asarray(0.5, dtype=arf)
with allow_input(interval=True, float=True):
    np.asarray(0.5, dtype=arf)

Randomness

Since python-flint does not support random number generation, the random module just uses np.random. Therefore, the return values may not be random up to the precision of arb, acb.

What it does

• This package adds a flarray which subclasses ndarray in order to
  • Override __array_namespace__ to numpy_flint_arb.np
  • Override dtype to return newly added _fl_dtype private attribute, since the actual internal dtype object cannot be overridden.
  • Override __array_finalize__ as recommended by the NumPy docs to return flarray with proper _fl_dtype instead of ndarray after Numpy operations.
• Partially supports linalg and (scipy.)special functions.
• Adds tomat() and frommat() to treat flarray as array of arb_mat / acb_mat, so that we can perform matrix operations like np.linalg.solve on flarray.
• Does not perform any parallelization to avoid complexity and to fully utilize the great python-flint library
  • Using arb_series and acb_series may be faster for additions but this is too hacky.
  • Defining custom dtype is way too complicated
  • Writing C extension would be theoretically also possible but is still too complicated.
• Does not support in operator since it tries to convert the return value to bool. Use newly added np.contains(x, y) and np.overlaps(x, y) instead.

Contributors ✨

Thanks goes to these wonderful people (emoji key):

34j 💻 🤔 📖

This project follows the all-contributors specification. Contributions of any kind welcome!

Credits

This package was created with Copier and the browniebroke/pypackage-template project template.