Skip to content

Commit 45b0074

Browse files
committed
chore: fix linting issues
1 parent 72e3f9e commit 45b0074

2 files changed

Lines changed: 22 additions & 9 deletions

File tree

pactus/utils/utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def encode_from_base256_with_type(hrp: str, typ: str, data: bytes) -> str:
2121

2222
def evaluate_polynomial(c: list[int], x: int, mod: int) -> int | None:
2323
"""
24-
Evaluates the polynomial f(x) = c[0] + c[1] * x + c[2] * x^2 + ... + c[n-1] * x^(n-1)
24+
Evaluate the polynomial f(x) = c[0] + c[1] * x + c[2] * x^2 + ... + c[n-1] * x^(n-1).
2525
2626
Args:
2727
c: List of polynomial coefficients (c[0] is the constant term)
@@ -30,6 +30,7 @@ def evaluate_polynomial(c: list[int], x: int, mod: int) -> int | None:
3030
3131
Returns:
3232
The computed value f(x) if success, None otherwise
33+
3334
"""
3435
if not c:
3536
return None
@@ -38,7 +39,7 @@ def evaluate_polynomial(c: list[int], x: int, mod: int) -> int | None:
3839
return c[0]
3940

4041
y = c[-1]
41-
for i in range(len(c)-2, -1, -1):
42+
for i in range(len(c) - 2, -1, -1):
4243
y = (y * x + c[i]) % mod
4344

4445
return y

tests/test_utils.py

Lines changed: 19 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,25 @@ def test_modulus_one(self):
3939

4040
def test_wikipedia_example(self):
4141
# https://en.wikipedia.org/wiki/Shamir%27s_secret_sharing
42-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 1, 2 ** 127 - 1), 1494)
43-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 2, 2 ** 127 - 1), 1942)
44-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 3, 2 ** 127 - 1), 2578)
45-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 4, 2 ** 127 - 1), 3402)
46-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 5, 2 ** 127 - 1), 4414)
47-
self.assertEqual(utils.evaluate_polynomial([1234, 166, 94], 6, 2 ** 127 - 1), 5614)
42+
self.assertEqual(
43+
utils.evaluate_polynomial([1234, 166, 94], 1, 2**127 - 1), 1494
44+
)
45+
self.assertEqual(
46+
utils.evaluate_polynomial([1234, 166, 94], 2, 2**127 - 1), 1942
47+
)
48+
self.assertEqual(
49+
utils.evaluate_polynomial([1234, 166, 94], 3, 2**127 - 1), 2578
50+
)
51+
self.assertEqual(
52+
utils.evaluate_polynomial([1234, 166, 94], 4, 2**127 - 1), 3402
53+
)
54+
self.assertEqual(
55+
utils.evaluate_polynomial([1234, 166, 94], 5, 2**127 - 1), 4414
56+
)
57+
self.assertEqual(
58+
utils.evaluate_polynomial([1234, 166, 94], 6, 2**127 - 1), 5614
59+
)
4860

4961

5062
if __name__ == "__main__":
51-
unittest.main()
63+
unittest.main()

0 commit comments

Comments
 (0)