def test_assorted_jac_mult() -> None:
ec = ec23_31
H = second_generator(ec)
HJ = _jac_from_aff(H)
for k1 in range(ec.n):
K1J = _mult(k1, ec.GJ, ec)
for k2 in range(ec.n):
K2J = _mult(k2, HJ, ec)
shamir = _double_mult(k1, ec.GJ, k2, ec.GJ, ec)
assert ec.is_on_curve(ec._aff_from_jac(shamir))
assert ec._jac_equality(shamir, _mult(k1 + k2, ec.GJ, ec))
shamir = _double_mult(k1, INFJ, k2, HJ, ec)
assert ec.is_on_curve(ec._aff_from_jac(shamir))
assert ec._jac_equality(shamir, K2J)
shamir = _double_mult(k1, ec.GJ, k2, INFJ, ec)
assert ec.is_on_curve(ec._aff_from_jac(shamir))
assert ec._jac_equality(shamir, K1J)
shamir = _double_mult(k1, ec.GJ, k2, HJ, ec)
assert ec.is_on_curve(ec._aff_from_jac(shamir))
K1JK2J = ec._add_jac(K1J, K2J)
assert ec._jac_equality(K1JK2J, shamir)
k3 = 1 + secrets.randbelow(ec.n - 1)
K3J = _mult(k3, ec.GJ, ec)
K1JK2JK3J = ec._add_jac(K1JK2J, K3J)
assert ec.is_on_curve(ec._aff_from_jac(K1JK2JK3J))
boscoster = _multi_mult([k1, k2, k3], [ec.GJ, HJ, ec.GJ], ec)
assert ec.is_on_curve(ec._aff_from_jac(boscoster))
assert ec._aff_from_jac(K1JK2JK3J) == ec._aff_from_jac(
boscoster), k3
assert ec._jac_equality(K1JK2JK3J, boscoster)
k4 = 1 + secrets.randbelow(ec.n - 1)
K4J = _mult(k4, HJ, ec)
K1JK2JK3JK4J = ec._add_jac(K1JK2JK3J, K4J)
assert ec.is_on_curve(ec._aff_from_jac(K1JK2JK3JK4J))
points = [ec.GJ, HJ, ec.GJ, HJ]
boscoster = _multi_mult([k1, k2, k3, k4], points, ec)
assert ec.is_on_curve(ec._aff_from_jac(boscoster))
assert ec._aff_from_jac(K1JK2JK3JK4J) == ec._aff_from_jac(
boscoster), k4
assert ec._jac_equality(K1JK2JK3JK4J, boscoster)
assert ec._jac_equality(K1JK2JK3J,
_multi_mult([k1, k2, k3, 0], points, ec))
assert ec._jac_equality(K1JK2J,
_multi_mult([k1, k2, 0, 0], points, ec))
assert ec._jac_equality(K1J, _multi_mult([k1, 0, 0, 0], points,
ec))
assert ec._jac_equality(INFJ, _multi_mult([0, 0, 0, 0], points,
ec))
err_msg = "mismatch between number of scalars and points: "
with pytest.raises(BTClibValueError, match=err_msg):
_multi_mult([k1, k2, k3, k4], [ec.GJ, HJ, ec.GJ], ec)
err_msg = "negative coefficient: "
with pytest.raises(BTClibValueError, match=err_msg):
_multi_mult([k1, k2, -k3], [ec.GJ, HJ, ec.GJ], ec)
with pytest.raises(BTClibValueError, match="negative first coefficient: "):
_double_mult(-5, HJ, 1, ec.GJ, ec)
with pytest.raises(BTClibValueError,
match="negative second coefficient: "):
_double_mult(1, HJ, -5, ec.GJ, ec)
random.seed(42)
# setup
us = []
vs = []
QJs = []
for _ in range(500):
us.append(random.getrandbits(ec.nlen) % ec.n)
vs.append(random.getrandbits(ec.nlen) % ec.n)
q = random.getrandbits(ec.nlen) % ec.n
QJs.append(_mult(q, ec.GJ, ec))
"""
for u, v, QJ in zip(us, vs, QJs):
t1 = ec._add_jac(_mult(u, ec.GJ, ec), _mult(v, QJ, ec))
t2 = _double_mult(u, ec.GJ, v, QJ, ec)
assert ec._jac_equality(t1, t2)
"""
start = time.time()
for u, v, QJ in zip(us, vs, QJs):
ec._add_jac(_mult(u, ec.GJ, ec), _mult(v, QJ, ec))
elapsed1 = time.time() - start
start = time.time()
for u, v, QJ in zip(us, vs, QJs):
_double_mult(u, ec.GJ, v, QJ, ec)
elapsed2 = time.time() - start
print(f"{elapsed2 / elapsed1:.0%}")