35 lines
1.4 KiB
Python
35 lines
1.4 KiB
Python
|
from sympy.ntheory.modular import crt, crt1, crt2, solve_congruence
|
||
|
|
from sympy.testing.pytest import raises
|
||
|
|
|
||
|
|
|
||
|
|
def test_crt():
|
||
|
|
def mcrt(m, v, r, symmetric=False):
|
||
|
|
assert crt(m, v, symmetric)[0] == r
|
||
|
|
mm, e, s = crt1(m)
|
||
|
|
assert crt2(m, v, mm, e, s, symmetric) == (r, mm)
|
||
|
|
|
||
|
|
mcrt([2, 3, 5], [0, 0, 0], 0)
|
||
|
|
mcrt([2, 3, 5], [1, 1, 1], 1)
|
||
|
|
|
||
|
|
mcrt([2, 3, 5], [-1, -1, -1], -1, True)
|
||
|
|
mcrt([2, 3, 5], [-1, -1, -1], 2*3*5 - 1, False)
|
||
|
|
|
||
|
|
assert crt([656, 350], [811, 133], symmetric=True) == (-56917, 114800)
|
||
|
|
|
||
|
|
|
||
|
|
def test_modular():
|
||
|
|
assert solve_congruence(*list(zip([3, 4, 2], [12, 35, 17]))) == (1719, 7140)
|
||
|
|
assert solve_congruence(*list(zip([3, 4, 2], [12, 6, 17]))) is None
|
||
|
|
assert solve_congruence(*list(zip([3, 4, 2], [13, 7, 17]))) == (172, 1547)
|
||
|
|
assert solve_congruence(*list(zip([-10, -3, -15], [13, 7, 17]))) == (172, 1547)
|
||
|
|
assert solve_congruence(*list(zip([-10, -3, 1, -15], [13, 7, 7, 17]))) is None
|
||
|
|
assert solve_congruence(
|
||
|
|
*list(zip([-10, -5, 2, -15], [13, 7, 7, 17]))) == (835, 1547)
|
||
|
|
assert solve_congruence(
|
||
|
|
*list(zip([-10, -5, 2, -15], [13, 7, 14, 17]))) == (2382, 3094)
|
||
|
|
assert solve_congruence(
|
||
|
|
*list(zip([-10, 2, 2, -15], [13, 7, 14, 17]))) == (2382, 3094)
|
||
|
|
assert solve_congruence(*list(zip((1, 1, 2), (3, 2, 4)))) is None
|
||
|
|
raises(
|
||
|
|
ValueError, lambda: solve_congruence(*list(zip([3, 4, 2], [12.1, 35, 17]))))
|