isqrt 0.12.0

Efficient pure Python implementation of the integer square root function.

Efficient pure Python implementation of the integer square root function.

Purpose

Given an arbitrarily large non-negative integer n, the integer square root function finds the largest integer r such that r**2 <= n and (r + 1)**2 > n.

The built-in math.isqrt function was introduced in Python 3.8 and should normally be used instead of the function defined in this library. To provide the best performance possible while retaining backwards-compatible behavior for this library, the implementation in this library invokes math.isqrt when it is available. If math.isqrt is not available, this library attempts to use math.sqrt and then (if math.sqrt does not produce the correct result or the input is outside the range supported by math.sqrt) defaults to a pure Python implementation in which the number of executed Python arithmetic operations is linear in the bit length of the input integer.

Installation and Usage

This library is available as a package on PyPI:

python -m pip install isqrt

The library can be imported in the usual way:

from isqrt import isqrt

The function isqrt provides an efficient pure Python implementation of the integer square root algorithm:

>>> isqrt(4)
2
>>> isqrt(16)
4
>>> list(map(isqrt, range(16, 26)))
[4, 4, 4, 4, 4, 4, 4, 4, 4, 5]
>>> isqrt(2**30000) == 2**15000
True

Development

All installation and development dependencies are managed using setuptools and are fully specified in setup.py. The extras_require parameter is used to specify optional requirements for various development tasks. This makes it possible to specify additional options (such as test, lint, and so on) when performing installation using pip:

python -m pip install .[test,lint]

Testing and Conventions

All unit tests are executed and their coverage is measured when using pytest (see setup.cfg for configuration details):

python -m pip install .[test]
python -m pytest

Alternatively, all unit tests are included in the module itself and can be executed using doctest:

python isqrt/isqrt.py -v

Style conventions are enforced using Pylint:

python -m pip install .[lint]
python -m pylint isqrt

Acknowledgments

The initial version of this function was posted on Stack Overflow. A more efficient version was implemented by Alexander Gosselin. The implementation in this package is adapted directly from these previous implementations.

Contributions

In order to contribute to the source code, open an issue or submit a pull request on the GitHub page for this library.

Versioning

Beginning with version 0.10.0, the version number format for this library and the changes to the library associated with version number increments conform with Semantic Versioning 2.0.0.

Publishing

This library can be published as a package on PyPI by a package maintainer. First, install the dependencies required for packaging and publishing:

python -m pip install .[publish]

Remove any old build/distribution files. Then, package the source into a distribution archive using the wheel package:

rm -rf dist *.egg-info
python setup.py sdist bdist_wheel

Finally, upload the package distribution archive to PyPI using the twine package:

python -m twine upload dist/*