From 9ff4b5d77bb4d8af91d0b3bbde6d652f12b8d537 Mon Sep 17 00:00:00 2001 From: Praveen Kumar Date: Fri, 26 Jun 2026 21:33:45 +0530 Subject: [PATCH 1/2] Add doctests to binary_exponentiation in fermat_little_theorem --- maths/fermat_little_theorem.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/maths/fermat_little_theorem.py b/maths/fermat_little_theorem.py index 4a3ecd05ce91..860d9e43cf96 100644 --- a/maths/fermat_little_theorem.py +++ b/maths/fermat_little_theorem.py @@ -4,8 +4,17 @@ # Here we assume that p is a prime number, b divides a, and p doesn't divide b # Wikipedia reference: https://en.wikipedia.org/wiki/Fermat%27s_little_theorem - def binary_exponentiation(a: int, n: float, mod: int) -> int: + """ + Compute (a^n) % mod using binary exponentiation. + + >>> binary_exponentiation(2, 5, 13) + 6 + >>> binary_exponentiation(3, 4, 5) + 1 + >>> binary_exponentiation(10, 0, 7) + 1 + """ if n == 0: return 1 From 1ba675efeb659f3ca8e3d55853aa56b85b66dcaf Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 26 Jun 2026 16:09:13 +0000 Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- maths/fermat_little_theorem.py | 1 + 1 file changed, 1 insertion(+) diff --git a/maths/fermat_little_theorem.py b/maths/fermat_little_theorem.py index 860d9e43cf96..76d111f6800d 100644 --- a/maths/fermat_little_theorem.py +++ b/maths/fermat_little_theorem.py @@ -4,6 +4,7 @@ # Here we assume that p is a prime number, b divides a, and p doesn't divide b # Wikipedia reference: https://en.wikipedia.org/wiki/Fermat%27s_little_theorem + def binary_exponentiation(a: int, n: float, mod: int) -> int: """ Compute (a^n) % mod using binary exponentiation.