test_all.py

from __future__ import annotations
from typing import Any, Callable, Sequence, TypeVar, Iterable, Protocol, TYPE_CHECKING

import math
import operator
import pickle
import platform
import random

import flint
import flint.typing as typ
import flint.flint_base.flint_base as flint_base
from flint.utils.flint_exceptions import DomainError, IncompatibleContextError


PYPY = platform.python_implementation() == "PyPy"

ctx = flint.ctx

def raises(f, exception) -> bool:
    try:
        f()
    except exception:
        return True
    return False


if TYPE_CHECKING:
    from typing import TypeIs


Tscalar = TypeVar('Tscalar', bound=flint_base.flint_scalar)
Tscalar_co = TypeVar('Tscalar_co', bound=flint_base.flint_scalar, covariant=True)
Tmpoly = TypeVar('Tmpoly', bound=flint_base.flint_mpoly)
Tmpoly_p = TypeVar('Tmpoly_p', bound=typ.mpoly_p)
Tmpolyctx_co = TypeVar('Tmpolyctx_co', bound=flint_base.flint_mpoly_context, covariant=True)
Tmpolyctx_p_co = TypeVar('Tmpolyctx_p_co', bound=typ.mpoly_context_p, covariant=True)


_default_ctx_string = """\
pretty = True      # pretty-print repr() output
unicode = False    # use unicode characters in output
prec = 53          # real/complex precision (in bits)
dps = 15           # real/complex precision (in digits)
cap = 10           # power series precision
threads = 1        # max number of threads used internally\
"""

def test_pyflint():

    assert flint.__version__ == "0.8.0"

    ctx = flint.ctx
    assert str(ctx) == repr(ctx) == _default_ctx_string
    assert ctx.prec == 53
    oldprec = ctx.prec
    try:
        f1 = flint.arb(2).sqrt()
        ctx.prec = 10
        f2 = flint.arb(2).sqrt()
        assert f1.rad() < f2.rad()
        assert 1e-17 < f1.rad() < 1e-15
        assert 1e-3 < f2.rad() < 1e-2
    finally:
        ctx.prec = oldprec

    assert ctx.dps == 15
    olddps = ctx.dps
    try:
        f1 = flint.arb(2).sqrt()
        ctx.dps = 3
        f2 = flint.arb(2).sqrt()
        assert f1.rad() < f2.rad()
        assert 1e-17 < f1.rad() < 1e-15
        assert 1e-4 < f2.rad() < 1e-3
    finally:
        ctx.dps = olddps

    assert ctx.cap == 10
    oldcap = ctx.cap
    try:
        f1 = flint.fmpz_series([1,1])
        ctx.cap = 5
        f2 = flint.fmpz_series([1,1])
        assert f1._equal_repr(flint.fmpz_series([1,1],10))
        assert f2._equal_repr(flint.fmpz_series([1,1],5))
    finally:
        ctx.cap = oldcap

    assert raises(lambda: setattr(ctx, "cap", -1), ValueError)
    assert raises(lambda: setattr(ctx, "prec", -1), ValueError)
    assert raises(lambda: setattr(ctx, "dps", -1), ValueError)


def test_showgood():
    from flint import good, showgood


def test_fmpz():
    assert flint.fmpz() == flint.fmpz(0)
    L = [0, 1, 2, 3, 2**31-1, 2**31, 2**63-1, 2**63, 2**64-1, 2**64]
    L += [-x for x in L]
    for i in L:
        f = flint.fmpz(i)
        assert int(f) == i
        assert flint.fmpz(f) == f
        assert flint.fmpz(str(i)) == f
    assert raises(lambda: flint.fmpz(1,2), TypeError) # type: ignore
    assert raises(lambda: flint.fmpz("qwe"), ValueError)
    assert raises(lambda: flint.fmpz([]), TypeError) # type: ignore
    for s in L:
        for t in L:
            for ltype in (flint.fmpz, int):
                for rtype in (flint.fmpz, int):

                    assert (ltype(s) == rtype(t)) == (s == t)
                    assert (ltype(s) != rtype(t)) == (s != t)
                    assert (ltype(s) < rtype(t)) == (s < t)
                    assert (ltype(s) <= rtype(t)) == (s <= t)
                    assert (ltype(s) > rtype(t)) == (s > t)
                    assert (ltype(s) >= rtype(t)) == (s >= t)

                    assert ltype(s) + rtype(t) == s + t
                    assert ltype(s) - rtype(t) == s - t
                    assert ltype(s) * rtype(t) == s * t
                    assert ltype(s) & rtype(t) == s & t
                    assert ltype(s) | rtype(t) == s | t
                    assert ltype(s) ^ rtype(t) == s ^ t
                    assert ~ltype(s) == ~s

                    if t == 0:
                        assert raises(lambda: ltype(s) // rtype(t), ZeroDivisionError)
                        assert raises(lambda: ltype(s) % rtype(t), ZeroDivisionError)
                        assert raises(lambda: divmod(ltype(s), rtype(t)), ZeroDivisionError)
                    else:
                        assert ltype(s) // rtype(t) == s // t
                        assert ltype(s) % rtype(t) == s % t
                        assert divmod(ltype(s), rtype(t)) == divmod(s, t)

                    if 0 <= t < 10:
                        assert (ltype(s) ** rtype(t)) == (s ** t)
                        assert ltype(s) << rtype(t) == s << t
                        assert ltype(s) >> rtype(t) == s >> t
                    elif -10 <= t < 0:
                        assert raises(lambda: ltype(s) << rtype(t), ValueError)
                        assert raises(lambda: ltype(s) >> rtype(t), ValueError)

    assert 2 ** flint.fmpz(2) == 4
    assert type(2 ** flint.fmpz(2)) == flint.fmpz
    assert raises(lambda: () ** flint.fmpz(1), TypeError) # type: ignore
    assert raises(lambda: flint.fmpz(1) ** (), TypeError) # type: ignore
    assert raises(lambda: flint.fmpz(1) ** -1, ValueError)

    mega = flint.fmpz(2) ** 8000000
    assert raises(lambda: mega ** mega, OverflowError)

    pow_mod_examples = [
        (2, 2, 3, 1),
        (2, -1, 5, 3),
        (2, 0, 5, 1),
        (2, 5, 1000, 32),
    ]
    for a, b, c, ab_mod_c in pow_mod_examples:
        assert pow(a, b, c) == ab_mod_c
        assert pow(flint.fmpz(a), b, c) == ab_mod_c
        assert pow(flint.fmpz(a), flint.fmpz(b), c) == ab_mod_c
        assert pow(flint.fmpz(a), b, flint.fmpz(c)) == ab_mod_c
        assert pow(flint.fmpz(a), flint.fmpz(b), flint.fmpz(c)) == ab_mod_c

        # 3-arg pow cannot be made to work with fmpz on PyPy
        # https://github.com/flintlib/python-flint/issues/74
        if not PYPY:
            # type checkers cannot understand 3-arg pow...
            assert pow(a, flint.fmpz(b), c) == ab_mod_c # type: ignore
            assert pow(a, b, flint.fmpz(c)) == ab_mod_c # type: ignore
            assert pow(a, flint.fmpz(b), flint.fmpz(c)) == ab_mod_c # type: ignore

    assert raises(lambda: pow(flint.fmpz(2), 2, 0), ValueError)
    # XXX: Handle negative modulus like int?
    assert raises(lambda: pow(flint.fmpz(2), 2, -1), ValueError)

    assert raises(lambda: pow(flint.fmpz(2), "asd", 2), TypeError) # type: ignore
    assert raises(lambda: pow(flint.fmpz(2), 2, "asd"), TypeError) # type: ignore

    f = flint.fmpz(2)
    assert f.numerator == f
    assert type(f.numerator) is flint.fmpz
    assert f.denominator == 1
    assert type(f.denominator) is flint.fmpz

    assert int(f) == 2
    assert type(int(f)) is int
    assert operator.index(f) == 2
    assert type(operator.index(f)) is int
    assert float(f) == 2.0
    assert type(float(f)) is float
    assert round(f) == 2
    assert type(round(f)) is flint.fmpz
    assert round(f, 1) == 2
    assert type(round(f, 1)) is flint.fmpz
    assert round(f, -1) == 0
    assert type(round(f, -1)) is flint.fmpz
    assert math.trunc(f) == 2
    assert type(math.trunc(f)) is flint.fmpz
    assert math.floor(f) == 2
    assert type(math.floor(f)) is flint.fmpz
    assert math.ceil(f) == 2
    assert type(math.ceil(f)) is flint.fmpz

    assert flint.fmpz(2) != []
    assert +flint.fmpz(0) == 0
    assert +flint.fmpz(1) == 1
    assert +flint.fmpz(-1) == -1
    assert -flint.fmpz(0) == 0
    assert -flint.fmpz(1) == -1
    assert -flint.fmpz(-1) == 1
    assert abs(flint.fmpz(0)) == 0
    assert abs(flint.fmpz(1)) == 1
    assert abs(flint.fmpz(-1)) == 1

    assert bool(flint.fmpz(0)) is False
    assert bool(flint.fmpz(1)) is True

    assert flint.fmpz(2).bit_length() == 2
    assert flint.fmpz(-2).bit_length() == 2
    assert flint.fmpz(2).height_bits() == 2
    assert flint.fmpz(-2).height_bits() == 2
    assert flint.fmpz(2).height_bits(signed=True) == 2
    assert flint.fmpz(-2).height_bits(signed=True) == -2

    f1 = flint.fmpz(1)
    f2 = flint.fmpz(2)
    f3 = flint.fmpz(3)
    f8 = flint.fmpz(8)

    assert f2 << 2 == 8
    assert f2 << f2 == 8
    assert 2 << f2 == 8
    assert raises(lambda: f2 << -1, ValueError)
    assert raises(lambda: 2 << -f1, ValueError)

    assert f8 >> 2 == f2
    assert f8 >> f2 == f2
    assert 8 >> f2 == f2
    assert raises(lambda: f2 >> -1, ValueError)
    assert raises(lambda: 2 >> -f1, ValueError)

    assert f2 & 3 == 2
    assert f2 & f3 == 2
    assert 2 & f3 == 2
    assert f2 | 3 == 3
    assert f2 | f3 == 3
    assert 2 | f3 == 3
    assert f2 ^ 3 == 1
    assert f2 ^ f3 == 1
    assert 2 ^ f3 == 1

    assert raises(lambda: f2 << (), TypeError) # type: ignore
    assert raises(lambda: () << f2, TypeError) # type: ignore
    assert raises(lambda: f2 >> (), TypeError) # type: ignore
    assert raises(lambda: () >> f2, TypeError) # type: ignore
    assert raises(lambda: f2 & (), TypeError) # type: ignore
    assert raises(lambda: () & f2, TypeError) # type: ignore
    assert raises(lambda: f2 | (), TypeError) # type: ignore
    assert raises(lambda: () | f2, TypeError) # type: ignore
    assert raises(lambda: f2 ^ (), TypeError) # type: ignore
    assert raises(lambda: () ^ f2, TypeError) # type: ignore

    ell = [1, 2, 3]
    ell[flint.fmpz(1)] = -2
    assert ell == [1, -2, 3]
    d = {flint.fmpz(2): 3}
    d[flint.fmpz(2)] = -1

    assert d == {flint.fmpz(2): -1}
    ctx.pretty = False
    assert repr(flint.fmpz(0)) == "fmpz(0)"
    assert repr(flint.fmpz(-27)) == "fmpz(-27)"
    ctx.pretty = True
    assert repr(flint.fmpz(0)) == "0"
    assert repr(flint.fmpz(-27)) == "-27"
    bigstr = '1' * 100
    big = flint.fmpz(bigstr)
    assert big.str() == bigstr
    assert big.str(condense=10) == '1111111111{...80 digits...}1111111111'

def test_fmpz_factor():
    assert flint.fmpz(6).gcd(flint.fmpz(9)) == 3
    assert flint.fmpz(6).gcd(9) == 3
    assert raises(lambda: flint.fmpz(2).gcd('asd'), TypeError) # type: ignore
    assert flint.fmpz(6).lcm(flint.fmpz(9)) == 18
    assert flint.fmpz(6).lcm(9) == 18
    assert raises(lambda: flint.fmpz(2).lcm('asd'), TypeError) # type: ignore
    assert flint.fmpz(25).factor() == [(5, 2)]
    n = flint.fmpz(10**100 + 1)
    assert n.factor() == [
        (73, 1), (137, 1), (401, 1), (1201, 1), (1601, 1), (1676321, 1), (5964848081, 1),
        (129694419029057750551385771184564274499075700947656757821537291527196801, 1)]
    assert n.factor(trial_limit=100) == [
        (73, 1), (137, 1), (401, 1),
        (2493516234411471571047384039650897753117456334167082044912715710972543643391271845384040149601, 1)]
    assert n.factor_smooth() == [
        (73, 1), (137, 1), (401, 1), (1201, 1), (1601, 1),
        (1296814508839693536173209832765271992846610925502473758289451540212712414540699659186801, 1)]

def test_fmpz_functions():
    T, F, VE, OE, DE = True, False, ValueError, OverflowError, DomainError
    cases = [
        # (f, [f(-1), f(0), f(1), f(2), ... f(10)]),
        (lambda n: flint.fmpz(n).is_prime(),
            [0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0]),
        (lambda n: flint.fmpz(n).is_probable_prime(),
            [0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 0, 0]),
        (lambda n: flint.fmpz(n).is_perfect_power(),
            [T, T, T, F, F, T, F, F, F, T, T, F]),
        (lambda n: flint.fmpz(n).is_square(),
            [F, T, T, F, F, T, F, F, F, F, T, F]),
        (lambda n: flint.fmpz(n).partitions_p(),
            [0, 1, 1, 2, 3, 5, 7, 11, 15, 22, 30, 42]),
        (lambda n: flint.fmpz(n).moebius_mu(),
            [1, 0, 1, -1, -1, 0, -1, 1, -1, 0, 0, 1]),
        (lambda n: flint.fmpz.fac_ui(n),
            [OE, 1, 1, 2, 6, 24, 120, 720, 5040, 40320, 362880, 3628800]),
        (lambda n: flint.fmpz.primorial_ui(n),
            [OE, 1, 1, 2, 6, 6, 30, 30, 210, 210, 210, 210, 2310, 2310]),
        (lambda n: flint.fmpz.fib_ui(n),
            [OE, 0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55]),
        (lambda n: flint.fmpz(n).rising(0),
            [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]),
        (lambda n: flint.fmpz(n).rising(1),
            [-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]),
        (lambda n: flint.fmpz(n).rising(2),
            [0, 0, 2, 6, 12, 20, 30, 42, 56, 72, 90, 110]),
        (lambda n: flint.fmpz.bin_uiui(n, 0),
            [OE, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]),
        (lambda n: flint.fmpz.bin_uiui(n, 1),
            [OE, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]),
        (lambda n: flint.fmpz.bin_uiui(n, 2),
            [OE, 0, 0, 1, 3, 6, 10, 15, 21, 28, 36, 45]),
        (lambda n: flint.fmpz.bell_number(n),
            [OE, 1, 1, 2, 5, 15, 52, 203, 877, 4140, 21147, 115975]),
        (lambda n: flint.fmpz.euler_number(n),
            [OE, 1, 0, -1, 0, 5, 0, -61, 0, 1385, 0, -50521]),
        (lambda n: flint.fmpz.stirling_s1(n, 1),
            [OE, 0, 1, -1, 2, -6, 24, -120, 720, -5040, 40320, -362880]),
        (lambda n: flint.fmpz.stirling_s2(n, 2),
            [OE, 0, 0, 1, 3, 7, 15, 31, 63, 127, 255, 511]),
        (lambda n: flint.fmpz(n).divisor_sigma(2),
            [1, 0, 1, 5, 10, 21, 26, 50, 50, 85, 91, 130]),
        (lambda n: flint.fmpz(n).euler_phi(),
            [0, 0, 1, 1, 2, 2, 4, 2, 6, 4, 6, 4]),
        (lambda n: flint.fmpz(n).isqrt(),
            [DE, 0, 1, 1, 1, 2, 2, 2, 2, 2, 3, 3]),
        (lambda n: flint.fmpz(n).sqrtrem(),
            [DE, (0, 0), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2), (2, 3), (2, 4), (3, 0), (3, 1)]),
        (lambda n: flint.fmpz(n).sqrtmod(11),
            [DE, 0, 1, DE, 5, 2, 4, DE, DE, DE, 3, DE]),
        (lambda n: flint.fmpz(n).root(3),
            [VE, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2]),
        (lambda n: flint.fmpz(n).jacobi(3),
            [-1, 0, 1, -1, 0, 1, -1, 0, 1, -1, 0, 1]),
        (lambda n: flint.fmpz(2).jacobi(n),
            [1, 1, 1, -1, -1, -1, -1, 1, 1, 1, 1, -1]),
    ]
    is_exception = lambda v: isinstance(v, type) and issubclass(v, Exception)

    for func, values in cases:
        for n, val in enumerate(values, -1):
            if is_exception(val):
                assert raises(lambda: func(n), val)
            else:
                assert func(n) == val

    assert raises(lambda: flint.fmpz(1).root(-1), ValueError)
    assert raises(lambda: flint.fmpz(1).jacobi('bad'), TypeError) # type: ignore

def test_fmpz_poly():
    Z = flint.fmpz_poly
    assert Z() == Z([])
    assert Z() == Z([0])
    assert Z() == Z([0,flint.fmpz(0),0])
    assert Z() == Z([0,0,0])
    assert Z() != Z([1])
    assert Z([1]) == Z([1])
    assert Z([1]) == Z([flint.fmpz(1)])
    assert Z(Z([1,2])) == Z([1,2])
    assert raises(lambda: Z([1,2,[]]), TypeError) # type: ignore
    assert raises(lambda: Z({}), TypeError) # type: ignore
    # XXX: This should probably be made to work:
    assert raises(lambda: Z((1,2,3)), TypeError) # type: ignore
    for ztype in [int, flint.fmpz]:
        assert Z([1,2,3]) + ztype(5) == Z([6,2,3])
        assert ztype(5) + Z([1,2,3]) == Z([6,2,3])
        assert Z([1,2,3]) - ztype(5) == Z([-4,2,3])
        assert ztype(5) - Z([1,2,3]) == Z([4,-2,-3])
        assert Z([1,2,3]) * ztype(5) == Z([5,10,15])
        assert ztype(5) * Z([1,2,3]) == Z([5,10,15])
        assert Z([11,6,2]) // ztype(5) == Z([2,1])
        assert ztype(5) // Z([-2]) == Z([-3])
        assert ztype(5) // Z([1,2]) == 0
        assert Z([11,6,2]) % ztype(5) == Z([1,1,2])
        assert ztype(5) % Z([-2]) == Z([-1])
        assert ztype(5) % Z([1,2]) == 5
        assert Z([1,2,3]) ** ztype(0) == 1
        assert Z([1,2,3]) ** ztype(1) == Z([1,2,3])
        assert Z([1,2,3]) ** ztype(2) == Z([1,4,10,12,9])
    assert divmod(Z([11,6,2]), Z([1,2])) == (Z([2,1]), Z([9,1]))
    assert raises(lambda: pow(Z([1,2]), 2, 3), NotImplementedError) # type: ignore
    assert +Z([1,2]) == Z([1,2])
    assert -Z([1,2]) == Z([-1,-2])
    assert raises(lambda: Z([1,2]) + [], TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) - [], TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) * [], TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) / [], TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) // [], TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) % [], TypeError) # type: ignore
    assert raises(lambda: divmod(Z([1,2]), []), TypeError) # type: ignore
    assert raises(lambda: [] + Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: [] - Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: [] * Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: [] / Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: [] // Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: [] % Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: divmod([], Z([1,2])), TypeError) # type: ignore
    assert raises(lambda: Z([1,2,3]) ** -1, DomainError) # type: ignore
    assert raises(lambda: Z([1,2,3]) ** Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) // Z([]), ZeroDivisionError)
    assert raises(lambda: Z([]) // Z([]), ZeroDivisionError)
    assert raises(lambda: Z([1,2]) % Z([]), ZeroDivisionError)
    assert raises(lambda: divmod(Z([1,2]), Z([])), ZeroDivisionError)
    assert raises(lambda: Z([1,2]) < Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) < [], TypeError) # type: ignore
    assert raises(lambda: [] < Z([1,2]), TypeError) # type: ignore
    assert Z([]).degree() == -1
    assert Z([]).length() == 0
    p = Z([1,2])
    assert p.length() == 2
    assert p.degree() == 1
    assert p[0] == 1
    assert p[1] == 2
    assert p[2] == 0
    assert p[-1] == 0
    assert raises(lambda: p.__setitem__(-1, 1), ValueError)
    p[0] = 3
    assert p[0] == 3
    p[4] = 7
    assert p.degree() == 4
    assert p[4] == 7
    assert p[3] == 0
    p[4] = 0
    assert p.degree() == 1
    assert p.coeffs() == [3,2]
    assert Z([]).coeffs() == []
    assert bool(Z([])) is False
    assert bool(Z([1])) is True
    ctx.pretty = False
    assert repr(Z([1,2])) == "fmpz_poly([1, 2])"
    ctx.pretty = True
    assert str(Z([1,2])) == "2*x + 1"
    assert str(Z([])) == "0"
    assert str(Z([1,0,2])) == "2*x^2 + 1"
    assert str(Z([-1,0,2])) == "2*x^2 + (-1)"
    assert str(Z([-1,0,1])) == "x^2 + (-1)"
    p = Z([3,4,5])
    assert p(2) == 31
    assert p(flint.fmpq(2,3)) == flint.fmpq(71,9)
    assert p(Z([1,-1])) == Z([12,-14,5])
    assert p(flint.fmpq_poly([2,3],5)) == flint.fmpq_poly([27,24,9],5)
    assert p(flint.arb("1.1")).overlaps(flint.arb("13.45"))
    assert p(flint.acb("1.1", "1.2")).overlaps(flint.acb("6.25", "18.00"))
    assert raises(lambda: p(None), TypeError) # type: ignore
    assert Z([1,2,3]).derivative() == Z([2,6])
    assert Z([1,2,-4]).height_bits() == 3
    assert Z([1,2,-4]).height_bits(signed=True) == -3
    assert Z([1,2,1]).sqrt() == Z([1,1])
    assert raises(lambda: Z([1,2,2]).sqrt(), DomainError)
    assert Z([1,0,2,0,3]).deflation() == (Z([1,2,3]), 2)
    assert Z([]).deflation() == (Z([]), 1)
    assert Z([1,1]).deflation() == (Z([1,1]), 1)
    [(r,m)] = Z([1,1]).complex_roots()
    assert m == 1
    assert r.overlaps(-1)
    assert Z([]).complex_roots() == []
    assert Z([1]).complex_roots() == []
    [(r,m)] = Z([1,1]).roots()
    assert m == 1
    assert r == -1
    assert Z([]).roots() == []
    assert Z([1]).roots() == []
    assert Z([1, 2]).roots() == []

def test_fmpz_poly_factor():
    Z = flint.fmpz_poly
    assert Z([1,2]).gcd(Z([3,4])) == 1
    assert Z([1,2,1]).gcd(Z([1,1])) == Z([1,1])
    assert raises(lambda: Z([1,2,1]).gcd([]), TypeError) # type: ignore
    assert Z([1,2,1]).factor() == (1, [(Z([1,1]), 2)])

def test_fmpz_poly_functions():
    Z = flint.fmpz_poly
    assert Z.cyclotomic(4) == Z([1,0,1])
    assert Z.cos_minpoly(10) == Z([-1,-1,1])
    assert Z.chebyshev_u(4) == Z([1,0,-12,0,16])
    assert Z.chebyshev_t(4) == Z([1,0,-8,0,8])
    assert Z.swinnerton_dyer(2) == Z([1,0,-10,0,1])
    assert Z.swinnerton_dyer(2, use_arb=False) == Z([1,0,-10,0,1])
    assert raises(lambda: Z.swinnerton_dyer(21), OverflowError)
    assert Z.hilbert_class_poly(-7) == Z([3375,1])
    assert raises(lambda: Z.hilbert_class_poly(-2), ValueError)
    assert Z([1]).is_cyclotomic() == 0
    assert Z([-1,1]).is_cyclotomic() == 1
    assert Z([1,1]).is_cyclotomic() == 2
    assert Z([1,2]).is_cyclotomic() == 0
    assert Z([1,2,1]).is_cyclotomic() == 0
    assert Z([2,2,1,1]).is_cyclotomic() == 0
    assert Z([2,2,1]).is_cyclotomic() == 0
    assert Z([1,2,2]).is_cyclotomic() == 0
    assert Z([1,1,1]).is_cyclotomic() == 3

def test_fmpz_mat():
    M = flint.fmpz_mat
    a = M(2,3,[1,2,3,4,5,6])
    b = M(2,3,[4,5,6,7,8,9])
    assert a == a
    assert a == M(a)
    assert a != b
    assert a.nrows() == 2
    assert a.ncols() == 3
    assert a.entries() == [1,2,3,4,5,6]
    assert a.table() == [[1,2,3],[4,5,6]]
    assert (a + b).entries() == [5,7,9,11,13,15]
    assert (a - b).entries() == [-3,-3,-3,-3,-3,-3]
    assert a.transpose() == M(3,2,[1,4,2,5,3,6])
    assert raises(lambda: a + 1, TypeError)
    assert raises(lambda: 1 + a, TypeError)
    assert raises(lambda: a - 1, TypeError)
    assert raises(lambda: 1 - a, TypeError)
    # XXX: Maybe there should be a ShapeError or something?
    assert raises(a.det, ValueError)
    assert +a == a
    assert -a == M(2,3,[-1,-2,-3,-4,-5,-6])
    c = M(2,2,[1,2,3,4])
    assert c.det() == -2
    assert raises(lambda: a + c, ValueError)
    assert (a * 3).entries() == [3,6,9,12,15,18]
    assert (3 * a).entries() == [3,6,9,12,15,18]
    assert (a * flint.fmpz(3)).entries() == [3,6,9,12,15,18]
    assert (flint.fmpz(3) * a).entries() == [3,6,9,12,15,18]
    assert M.randrank(5,7,3,10).rank() == 3
    A = M.randbits(5,3,2)
    B = M.randtest(3,7,3)
    C = M.randtest(7,2,4)
    assert (A.nrows(),A.ncols()) == (5,3)
    assert (B.nrows(),B.ncols()) == (3,7)
    assert (C.nrows(),C.ncols()) == (7,2)
    assert A*(B*C) == (A*B)*C
    assert raises(lambda: A*C, ValueError)
    assert bool(M(2,2,[0,0,0,0])) is False
    assert bool(M(2,2,[0,0,0,1])) is True
    ctx.pretty = False
    assert repr(M(2,2,[1,2,3,4])) == 'fmpz_mat(2, 2, [1, 2, 3, 4])'
    ctx.pretty = True
    assert str(M(2,2,[1,2,3,4])) == '[1, 2]\n[3, 4]'
    assert M(1,2,[3,4]) * flint.fmpq(1,3) == flint.fmpq_mat(1, 2, [1, flint.fmpq(4,3)])
    assert flint.fmpq(1,3) * M(1,2,[3,4]) == flint.fmpq_mat(1, 2, [1, flint.fmpq(4,3)])
    assert raises(lambda: M(1,2,[3,4]) / 3, DomainError)
    assert M(1,2,[2,4]) / 2 == M(1,2,[1,2])
    assert M(2,2,[1,2,3,4]).inv().det() == flint.fmpq(1) / M(2,2,[1,2,3,4]).det()
    assert M(2,2,[1,2,3,4]).inv().inv() == M(2,2,[1,2,3,4])
    assert raises(lambda: M.randrank(4,3,4,1), ValueError)
    assert raises(lambda: M.randrank(3,4,4,1), ValueError)
    assert M(1,1,[3]) ** 5 == M(1,1,[3**5])
    assert raises(lambda: pow(M([[1]]), 2, 3), NotImplementedError)
    assert raises(lambda: M(1,2) ** 3, ValueError)
    assert raises(lambda: M(1,1) ** M(1,1), TypeError)
    assert raises(lambda: 1 ** M(1,1), TypeError)
    assert raises(lambda: M([1]), TypeError)
    assert raises(lambda: M([[1],[2,3]]), ValueError)
    assert raises(lambda: M(None), TypeError)
    assert raises(lambda: M(2,2,[1,2,3]), ValueError)
    assert raises(lambda: M(2,2,2,2), TypeError)
    assert M([[1,2,3],[4,5,6]]) == M(2,3,[1,2,3,4,5,6])
    assert raises(lambda: M([[1]]) < M([[2]]), TypeError)
    assert (M([[1]]) == 1) is False
    assert (1 == M([[1]])) is False
    assert (M([[1]]) != 1) is True
    assert (1 != M([[1]])) is True
    D = M([[1,2],[3,4]])
    assert (D[0,0],D[0,1],D[1,0],D[1,1]) == (1,2,3,4)
    D[0,0] = 3
    assert D == M([[3,2],[3,4]])

    def set_bad(i,j):
        D[i,j] = -1

    # XXX: Should be IndexError
    raises(lambda: set_bad(2,0), IndexError)
    raises(lambda: set_bad(0,2), IndexError)
    raises(lambda: D[0,2], IndexError)
    raises(lambda: D[0,2], IndexError)
    # XXX: Negative indices?
    raises(lambda: set_bad(-1,0), IndexError)
    raises(lambda: set_bad(0,-1), IndexError)
    raises(lambda: D[-1,0], IndexError)
    raises(lambda: D[0,-1], IndexError)
    assert M.hadamard(2) == M([[1,1],[1,-1]])
    assert raises(lambda: M.hadamard(3), ValueError)
    assert M.hadamard(2).is_hadamard() is True
    assert M([[1,2],[3,4]]).is_hadamard() is False
    M1 = M([[1,0],[1,1]])
    M2 = M([[1,0],[-1,1]])
    x = M([[3],[4]])
    b = M([[3],[7]])
    assert M1.inv() == M2
    assert M2.inv() == M1
    assert M1.inv(integer=True) == M2
    assert M2.inv(integer=True) == M1
    assert M1*x == b
    assert M1.solve(b) == x
    assert M1.solve(b, integer=True) == x
    assert raises(lambda: M1.solve([]), TypeError)
    assert raises(lambda: M1.solve(M([[1]])), ValueError)
    assert raises(lambda: M([[1,1],[1,1]]).solve(b), ZeroDivisionError)
    assert raises(lambda: M([[1,2],[3,4],[5,6]]).solve(b), ValueError)
    assert M([[1,0],[1,2]]).solve(b) == flint.fmpq_mat([[3],[2]])
    assert raises(lambda: M([[1,0],[1,2]]).solve(b, integer=True), ValueError)
    assert raises(lambda: M([[1,2,3],[4,5,6]]).inv(), ValueError)
    assert raises(lambda: M([[1,1],[1,1]]).inv(), ZeroDivisionError)
    assert raises(lambda: M([[1,0],[1,2]]).inv(integer=True), ValueError)
    half = flint.fmpq(1,2)
    assert M([[1,0],[1,2]]).inv() == flint.fmpq_mat([[1, 0], [-half, half]])
    M3 = M([[1,2,3],[4,5,6],[7,8,9]])
    M3_copy = M(M3)
    M3r = M([[-3,0,3],[0,-3,-6],[0,0,0]])
    assert M3.rref() == (M3r, -3, 2)
    assert M3 != M3r
    assert M3.rref(inplace=True) == (M3r, -3, 2)
    assert M3 == M3r
    M3 = M3_copy
    M3n = M([[3,0,0],[-6,0,0],[3,0,0]])
    assert M3.nullspace() == (M3n, 1)
    assert M3 * M3.nullspace()[0] == M(3,3,[0]*9)
    # XXX: lll core dumps on a singular matrix
    M4 = M([[1,2,3],[4,5,6],[7,8,10]])
    L4 = M([[0,0,1],[-1,1,0],[2,1,0]])
    T4 = M([[1,-2,1],[0,5,-3],[-2,1,0]])
    assert L4 == T4 * M4
    assert M4.lll() == L4
    assert M4.lll(transform=True) == (L4, T4)
    # XXX: rep="gram" consumes all memory in the system and core dumps
    #for rep in "zbasis", "gram":
    rep = "zbasis"
    for gram in "approx", "exact":
        assert M4.lll(rep=rep, gram=gram) == L4
        assert M4.lll(rep=rep, gram=gram, transform=True) == (L4, T4)
    assert raises(lambda: M4.lll(rep="bad"), ValueError)
    assert raises(lambda: M4.lll(gram="bad"), ValueError)
    M5 = M([[1,2,3],[4,5,6]])
    H5 = M([[1,2,3],[0,3,6]])
    T5 = M([[1,0],[4,-1]])
    assert H5 == T5 * M5
    assert M5.hnf() == H5
    assert M5.hnf(transform=True) == (H5, T5)
    assert M5.is_hnf() is False
    assert H5.is_hnf() is True
    S5 = M([[1,0,0],[0,3,0]])
    assert M5.snf() == S5
    assert M5.is_snf() is False
    assert S5.is_snf() is True
    M6 = M([[2,0,0],[0,2,1],[0,0,2]])
    assert M6.charpoly() == flint.fmpz_poly([-8,12,-6,1])
    assert M6.minpoly() == flint.fmpz_poly([4,-4,1])
    assert list(M6) == [2,0,0,0,2,1,0,0,2]


def test_fmpz_series():
    Zp = flint.fmpz_poly
    Z = flint.fmpz_series
    ctx = flint.ctx
    assert ctx.cap == 10
    s1 = Z([1,2])
    s2 = Z([1,2])
    s3 = Z([1,1])
    s4 = Z([1,2],11)
    s5 = Z([1,2],3)
    p1 = Zp([1,2])
    assert s1.prec == 10
    assert s4.prec == 11
    assert s5.prec == 3
    assert s1._equal_repr(s1) is True
    assert s1._equal_repr(s2) is True
    assert s1._equal_repr(s3) is False
    assert s1._equal_repr(s4) is False
    assert s1._equal_repr(p1) is False
    assert s1._equal_repr(flint.fmpz(1)) is False
    assert s1._equal_repr(1) is False
    assert Z([])._equal_repr(Z([0])) is True
    assert Z([1])._equal_repr(Z([0])) is False
    p = Z([1,2])
    assert p.prec == 10
    p4 = Z([1,2], prec=4)
    assert p4.prec == 4
    p4 = Z(p, prec=4)
    assert p4.prec == 4
    # XXX: this gives a core dump:
    # s = Z([1,2])
    # s[10**10] = 1
    assert Z(Z([1]))._equal_repr(Z([1]))
    assert Z(Zp([1]))._equal_repr(Z([1]))
    assert Z(1)._equal_repr(Z([1]))
    assert raises(lambda: Z([1]) < Z([1]), TypeError) # type: ignore
    assert len(Z([1,2])) == 2
    assert Z([1,2]).length() == 2
    s5 = Z([1,2])
    assert s5[1] == 2
    assert s5[2] == 0
    assert s5[-1] == 0
    # XXX: This goes beyond cap. Should it give an error?
    assert s5[100] == 0
    s5[2] = -1
    assert s5[2] == -1
    assert s5._equal_repr(Z([1,2,-1]))

    def set_bad():
        s5[-1] = 3

    assert raises(set_bad, ValueError)
    assert Z([1,2,0,4]).str() == "1 + 2*x + 4*x^3 + O(x^10)"
    assert Z([1,2,0,4]).repr() == "fmpz_series([1, 2, 0, 4], prec=10)"
    assert Z([],0).str() == "O(x^0)"
    assert Z([],-1).str() == "(invalid power series)"
    assert (+Z([1,2]))._equal_repr(Z([1,2]))
    assert (-Z([1,2]))._equal_repr(Z([-1,-2]))
    assert (Z([1,2]) + Z([3,4,5]))._equal_repr(Z([4,6,5]))
    assert (Z([1,2]) + 1)._equal_repr(Z([2,2]))
    assert (Z([1,2]) + Zp([3,4,5]))._equal_repr(Z([4,6,5]))
    assert (1 + Z([1,2]))._equal_repr(Z([2,2]))
    assert (Zp([1,2]) + Z([3,4,5]))._equal_repr(Z([4,6,5]))
    assert raises(lambda: Z([1,2]) + [], TypeError) # type: ignore
    assert raises(lambda: [] + Z([1,2]), TypeError) # type: ignore
    assert (Z([1,2]) - Z([3,5]))._equal_repr(Z([-2,-3]))
    assert (Z([1,2]) - 1)._equal_repr(Z([0,2]))
    assert (1 - Z([1,2]))._equal_repr(Z([0,-2]))
    assert raises(lambda: [] - Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) - [], TypeError) # type: ignore
    assert (Z([1,2]) * Z([1,2]))._equal_repr(Z([1,4,4]))
    assert (2 * Z([1,2]))._equal_repr(Z([2,4]))
    assert (Z([1,2]) * 2)._equal_repr(Z([2,4]))
    assert raises(lambda: [] * Z([1,2]), TypeError) # type: ignore
    assert raises(lambda: Z([1,2]) * [], TypeError) # type: ignore
    assert Z([1,2]).valuation() == 0
    assert Z([0,2]).valuation() == 1
    assert Z([0,0]).valuation() == -1
    assert (Z([1,1]) / Z([1,-1]))._equal_repr(Z([1,2,2,2,2,2,2,2,2,2]))
    assert raises(lambda: Z([1,1]) / Z([]), ZeroDivisionError)
    assert (Z([]) / Z([1,-1]))._equal_repr(Z([]))
    # quotient would not be a power series
    assert raises(lambda: Z([1,1]) / Z([0,1]), ValueError)
    # leading term in denominator is not a unit
    assert raises(lambda: Z([1,1]) / Z([2,1]), ValueError)
    assert (Z([0,1,1]) / Z([0,1,-1]))._equal_repr(Z([1,2,2,2,2,2,2,2,2],9))
    # XXX: This gives leading term not a unit:
    # assert Z([2,4]) / 2 == Z([1,2])
    assert (Z([1,1]) / -1)._equal_repr(Z([-1,-1]))
    assert raises(lambda: Z([1,1]) / [], TypeError) # type: ignore
    assert raises(lambda: [] / Z([1,1]), TypeError) # type: ignore
    assert (Z([1,2]) ** 2)._equal_repr(Z([1,4,4]))
    assert raises(lambda: pow(Z([1,2]), 3, 5), NotImplementedError) # type: ignore
    assert (Z([1,2])(Z([0,1,2])))._equal_repr(Z([1,2,4]))
    assert raises(lambda: Z([1,2])(Z([1,2])), ValueError)
    assert raises(lambda: Z([1,2])([]), TypeError) # type: ignore
    coeffs = [0, 1, -2, 8, -40, 224, -1344, 8448, -54912, 366080]
    assert Z([0,1,2]).reversion()._equal_repr(Z(coeffs))
    # power series reversion must have valuation 1
    assert raises(lambda: Z([1,1]).reversion(), ValueError)
    # leading term is not a unit
    assert raises(lambda: Z([0,2,1]).reversion(), ValueError)


def test_fmpq() -> None:
    Q = flint.fmpq
    assert Q() == Q(0)
    assert Q(0) != Q(1)
    assert Q(0) == 0
    assert 0 == Q(0)
    assert Q(2) != 1
    assert 1 != Q(2)
    assert Q(1) != ()
    assert Q(1,2) != 1
    assert Q(2,3) == Q(flint.fmpz(2),3)
    assert Q(-2,-4) == Q(1,2)
    assert Q("1") == Q(1)
    assert Q("1/2") == Q(1,2)
    assert Q(Q(1,2),Q(3,4)) == Q(2,3)
    assert Q(1,Q(2,3)) == Q(3,2)
    assert Q(Q(1,2),3) == Q(1,6)
    assert Q(flint.fmpz(1),Q(2,3)) == Q(3,2)
    assert Q(Q(1,2),flint.fmpz(3)) == Q(1,6)
    assert raises(lambda: Q("1.0"), ValueError)
    assert raises(lambda: Q("1.5"), ValueError)
    assert raises(lambda: Q("1/2/3"), ValueError)
    assert raises(lambda: Q([]), TypeError) # type: ignore
    assert raises(lambda: Q(1, []), TypeError) # type: ignore
    assert raises(lambda: Q([], 1), TypeError) # type: ignore
    assert bool(Q(0)) is False
    assert bool(Q(1)) is True
    assert Q(1,3) + Q(2,3) == 1
    assert Q(1,3) - Q(2,3) == Q(-1,3)
    assert Q(1,3) * Q(2,3) == Q(2,9)
    assert Q(1,3) + 2 == Q(7,3)
    assert 2 + Q(1,3) == Q(7,3)
    assert Q(1,3) - 2 == Q(-5,3)
    assert 2 - Q(1,3) == Q(5,3)
    assert Q(1,3) * 2 == Q(2,3)
    assert 2 * Q(1,3) == Q(2,3)
    assert Q(2,3) / Q(4,5) == Q(5,6)
    assert Q(2,3) / 5 == Q(2,15)
    assert Q(2,3) / flint.fmpz(5) == Q(2,15)
    assert 5 / Q(2,3) == Q(15,2)
    assert flint.fmpz(5) / Q(2,3) == Q(15,2)
    assert Q(2,3)/5 == Q(2,15)
    assert Q(1,2) ** 2 == Q(1,4)
    assert Q(1,2) ** -2 == Q(4)
    assert raises(lambda: Q(0) ** -1, ZeroDivisionError)
    assert raises(lambda: Q(1,2) ** Q(1,2), TypeError) # type: ignore
    assert raises(lambda: Q(1,2) ** [], TypeError) # type: ignore
    assert raises(lambda: [] ** Q(1,2), TypeError) # type: ignore
    assert raises(lambda: pow(Q(1,2),2,3), TypeError) # type: ignore

    megaz = flint.fmpz(2) ** 8000000
    megaq = Q(megaz)
    assert raises(lambda: megaq ** megaz, OverflowError)

    assert raises(lambda: Q(1,2) + [], TypeError) # type: ignore
    assert raises(lambda: Q(1,2) - [], TypeError) # type: ignore
    assert raises(lambda: Q(1,2) * [], TypeError) # type: ignore
    assert raises(lambda: Q(1,2) / [], TypeError) # type: ignore
    assert raises(lambda: [] + Q(1,2), TypeError) # type: ignore
    assert raises(lambda: [] - Q(1,2), TypeError) # type: ignore
    assert raises(lambda: [] * Q(1,2), TypeError) # type: ignore
    assert raises(lambda: [] / Q(1,2), TypeError) # type: ignore
    assert (Q(1,2) == 1) is False
    assert (Q(1,2) != 1) is True
    assert (Q(1,2) <  1) is True
    assert (Q(1,2) <= 1) is True
    assert (Q(1,2) >  1) is False
    assert (Q(1,2) >= 1) is False
    assert (Q(1,2) == Q(3,4)) is False
    assert (Q(1,2) != Q(3,4)) is True
    assert (Q(1,2) <  Q(3,4)) is True
    assert (Q(1,2) <= Q(3,4)) is True
    assert (Q(1,2) >  Q(3,4)) is False
    assert (Q(1,2) >= Q(3,4)) is False
    assert (Q(1,2) == Q(1,2)) is True
    assert (Q(1,2) != Q(1,2)) is False
    assert (Q(1,2) <  Q(1,2)) is False
    assert (Q(1,2) <= Q(1,2)) is True
    assert (Q(1,2) >  Q(1,2)) is False
    assert (Q(1,2) >= Q(1,2)) is True
    assert raises(lambda: Q(1,2) > [], TypeError) # type: ignore
    assert raises(lambda: [] < Q(1,2), TypeError) # type: ignore

    ctx.pretty = False
    assert repr(Q(-2,3)) == "fmpq(-2,3)"
    assert repr(Q(3)) == "fmpq(3)"
    ctx.pretty = True
    assert str(Q(-2,3)) == "-2/3"
    assert str(Q(3)) == "3"

    assert Q(2,3).p == Q(2,3).numer() == Q(2,3).numerator == 2
    assert Q(2,3).q == Q(2,3).denom() == Q(2,3).denominator == 3
    assert +Q(5,7) == Q(5,7)
    assert -Q(5,7) == Q(-5,7)
    assert -Q(-5,7) == Q(5,7)
    assert abs(Q(5,7)) == Q(5,7)
    assert abs(-Q(5,7)) == Q(5,7)
    assert raises(lambda: Q(1,0), ZeroDivisionError)
    assert raises(lambda: Q(1,2) / Q(0), ZeroDivisionError)
    assert raises(lambda: Q(1,2) / 0, ZeroDivisionError)

    assert Q(2,3).gcd(Q(4,9)) == Q(2,9)
    assert Q(2,3).gcd(5) == Q(1,3)
    assert raises(lambda: Q(2,3).gcd([]), TypeError) # type: ignore

    assert Q(5,3).floor() == flint.fmpz(1)
    assert Q(-5,3).floor() == flint.fmpz(-2)
    assert Q(5,3).ceil() == flint.fmpz(2)
    assert Q(-5,3).ceil() == flint.fmpz(-1)

    assert type(int(Q(5,3))) is int
    assert type(math.floor(Q(5,3))) is flint.fmpz
    assert type(math.ceil(Q(5,3))) is flint.fmpz
    assert type(math.trunc(Q(5,3))) is flint.fmpz
    assert type(round(Q(5,3))) is flint.fmpz
    assert type(round(Q(5,3))) is flint.fmpz
    assert type(round(Q(5,3), 0)) is flint.fmpq
    assert type(round(Q(5,3), 1)) is flint.fmpq

    assert int(Q(5,3)) == 1
    assert math.floor(Q(5,3)) == flint.fmpz(1)
    assert math.ceil(Q(5,3)) == flint.fmpz(2)
    assert math.trunc(Q(5,3)) == flint.fmpz(1)
    assert round(Q(5,3)) == flint.fmpz(2)

    assert int(Q(-5,3)) == flint.fmpz(-1)
    assert math.floor(Q(-5,3)) == flint.fmpz(-2)
    assert math.ceil(Q(-5,3)) == flint.fmpz(-1)
    assert math.trunc(Q(-5,3)) == flint.fmpz(-1)
    assert round(Q(-5,3)) == -2

    assert round(Q(100,3), 2) == Q(3333,100)
    assert round(Q(100,3), 0) == Q(33,1)
    assert round(Q(100,3), -1) == Q(30,1)
    assert round(Q(100,3), -2) == Q(0)

    d = {}
    d[Q(1,2)] = 3
    d[Q(1,2)] = 4
    assert d == {Q(1,2):4}

    assert Q(-5,3).height_bits() == 3
    assert Q(-5,3).height_bits(signed=True) == -3

    cases1: list[tuple[Callable[[flint.fmpq], flint.fmpq], list[flint.fmpq]]] = [
        (lambda q: q.next(),
            [Q(0), Q(1), Q(-1), Q(1,2), Q(-1,2), Q(2), Q(-2), Q(1,3), Q(-1,3), Q(3)]),
        (lambda q: q.next(signed=False),
            [Q(0), Q(1), Q(1,2), Q(2), Q(1,3), Q(3), Q(2,3), Q(3,2), Q(1,4), Q(4)]),
        (lambda q: q.next(minimal=False),
            [Q(0), Q(1), Q(-1), Q(1,2), Q(-1,2), Q(2), Q(-2), Q(1,3), Q(-1,3), Q(3,2)]),
        (lambda q: q.next(signed=False, minimal=False),
            [Q(0), Q(1), Q(1,2), Q(2), Q(1,3), Q(3,2), Q(2,3), Q(3), Q(1,4), Q(4,3)]),
    ]
    for func, values in cases1:
        for val1, val2 in zip(values[:-1], values[1:]):
            assert func(val1) == val2
    raises(lambda: Q(-1).next(signed=False), ValueError)

    OE = OverflowError
    cases2: list[tuple[Callable[[int], flint.fmpq], list[flint.fmpq | type[OverflowError]]]] = [
        (flint.fmpq.bernoulli,
            [OE, Q(1), Q(-1,2), Q(1,6), Q(0), Q(-1,30)]),
        (lambda n: flint.fmpq.bernoulli(n, cache=True),
            [OE, Q(1), Q(-1,2), Q(1,6), Q(0), Q(-1,30)]),
        (flint.fmpq.harmonic,
            [OE, Q(0), Q(1), Q(3,2), Q(11, 6), Q(25, 12)]),
        (lambda n: flint.fmpq.dedekind_sum(n, 3),
            [-Q(1,18), Q(0), Q(1,18), -Q(1,18), Q(0), Q(1,18), -Q(1,18)]),
    ]
    is_exception = lambda v: isinstance(v, type) and issubclass(v, Exception)

    for func2, values2 in cases2:
        for n, val in enumerate(values2, -1):
            if is_exception(val):
                assert raises(lambda: func2(n), val)
            else:
                assert func2(n) == val

def test_fmpq_poly():
    Q = flint.fmpq_poly
    Z = flint.fmpz_poly
    assert Q() == Q([]) == Q([0]) == Q([0,0])
    assert Q() != Q([1])
    assert Q([1]) == Q([1])
    assert Q([1], 2) == Q([flint.fmpq(1,2)])
    assert raises(lambda: Q([1,[]]), TypeError) # type: ignore
    assert raises(lambda: Q({}), TypeError) # type: ignore
    assert raises(lambda: Q([1], []), TypeError) # type: ignore
    assert raises(lambda: Q([1], 0), ZeroDivisionError)
    assert bool(Q()) is False
    assert bool(Q([1])) is True
    assert Q(Q([1,2])) == Q([1,2])
    assert Q(Z([1,2])) == Q([1,2])
    assert Q([1,2]) + 3 == Q([4,2])
    assert 3 + Q([1,2]) == Q([4,2])
    assert Q([1,2]) - 3 == Q([-2,2])
    assert 3 - Q([1,2]) == Q([2,-2])
    assert raises(lambda: Q([1]) + [], TypeError) # type: ignore
    assert raises(lambda: [] + Q([1]), TypeError) # type: ignore
    assert raises(lambda: Q([1]) - [], TypeError) # type: ignore
    assert raises(lambda: [] - Q([1]), TypeError) # type: ignore
    assert raises(lambda: Q([1]) * [], TypeError) # type: ignore
    assert raises(lambda: [] * Q([1]), TypeError) # type: ignore
    assert Q([1,2,1]) // Q([1,1]) == Q([1,1])
    assert Q([1,2,1]) % Q([1,1]) == 0
    assert divmod(Q([1,2,1]), Q([1,1])) == (Q([1,1]), 0)
    assert raises(lambda: Q([1,2,1]) // [], TypeError) # type: ignore
    assert raises(lambda: [] // Q([1,2,1]), TypeError) # type: ignore
    assert raises(lambda: Q([1,2,1]) % [], TypeError) # type: ignore
    assert raises(lambda: [] % Q([1,2,1]), TypeError) # type: ignore
    assert raises(lambda: divmod(Q([1,2,1]), []), TypeError) # type: ignore
    assert raises(lambda: divmod([], Q([1,2,1])), TypeError) # type: ignore
    assert raises(lambda: Q([1,2,1]) / 0, ZeroDivisionError)
    assert raises(lambda: Q([1,2,1]) // 0, ZeroDivisionError)
    assert raises(lambda: Q([1,2,1]) % 0, ZeroDivisionError)
    assert raises(lambda: divmod(Q([1,2,1]), 0), ZeroDivisionError)
    assert +Q([1,2]) == Q([1,2])
    assert -Q([1,2]) == Q([-1,-2])
    assert Q([flint.fmpq(1,2),1]) * 2 == Q([1,2])
    assert Q([1,2]) == Z([1,2])
    assert Z([1,2]) == Q([1,2])
    assert Q([1,2]) != Z([3,2])
    assert Z([1,2]) != Q([3,2])
    assert Q([1,2]) != []
    assert raises(lambda: Q([1,2]) < Q([1,2]), TypeError) # type: ignore
    assert Q([1,2,3])*Q([1,2]) == Q([1,4,7,6])
    assert Q([1,2,3])*Z([1,2]) == Q([1,4,7,6])
    assert Q([1,2,3]) * 3 == Q([3,6,9])
    assert 3 * Q([1,2,3]) == Q([3,6,9])
    assert Q([1,2,3]) * flint.fmpq(2,3) == (Q([1,2,3]) * 2) / 3
    assert flint.fmpq(2,3) * Q([1,2,3]) == (Q([1,2,3]) * 2) / 3
    assert Q([1,2]) / Q([1,2]) == Q([1])
    assert raises(lambda: Q([1,2]) / Q([2,2]), DomainError)
    assert Q([1,2,3]) / flint.fmpq(2,3) == Q([1,2,3]) * flint.fmpq(3,2)
    assert Q([1,2,3]) ** 2 == Q([1,2,3]) * Q([1,2,3])
    assert raises(lambda: pow(Q([1,2]), 3, 5), NotImplementedError) # type: ignore
    assert Q([1,2,flint.fmpq(1,2)]).coeffs() == [1,2,flint.fmpq(1,2)]
    assert Q().coeffs() == []
    assert Q().degree() == -1
    assert Q([1]).degree() == 0
    assert Q([1,2]).degree() == 1
    assert Q().length() == 0
    assert Q([1]).length() == 1
    assert Q([1,2]).length() == 2
    assert (Q([1,2,3]) / 5).numer() == (Q([1,2,3]) / 5).p == Z([1,2,3])
    assert (Q([1,2,3]) / 5).denom() == (Q([1,2,3]) / 5).q == 5
    ctx.pretty = False
    assert repr(Q([15,20,10])) == "fmpq_poly([15, 20, 10])"
    assert repr(Q([15,20,10]) / 25) == "fmpq_poly([3, 4, 2], 5)"
    ctx.pretty = True
    assert str(Q([3,4,2],5)) == "2/5*x^2 + 4/5*x + 3/5"
    a = Q([2,2,3],4)
    assert a[2] == flint.fmpq(3,4)
    assert a[-1] == flint.fmpq(0)
    a[2] = 4

    def set_bad():
        a[-1] = 2

    assert raises(set_bad, ValueError)
    assert a == Q([1,1,8],2)
    p = Q([3,4,5],7)
    assert p(2) == flint.fmpq(31,7)
    assert p(flint.fmpq(2,3)) == flint.fmpq(71,63)
    assert p(Z([1,-1])) == Q([12,-14,5],7)
    assert p(flint.fmpq_poly([2,3],5)) == flint.fmpq_poly([27,24,9],7*5)
    assert raises(lambda: p([]), TypeError) # type: ignore
    assert Q([1,2,3]).derivative() == Q([2,6])
    assert Q([1,2,3]).integral() == Q([0,1,1,1])
    assert Q([1,2,1]).gcd(Q([1,1])) == Q([1,1])
    assert Q([1,2,1]).xgcd(Q([1,1])) == (Q([1,1]), 0, 1)
    assert raises(lambda: Q([1,2,1]).gcd([]), TypeError) # type: ignore
    assert raises(lambda: Q([1,2,1]).xgcd([]), TypeError) # type: ignore
    assert Q([1,2,1]).factor() == (1, [(Q([1,1]), 2)])
    assert Q.bernoulli_poly(3) == Q([0,1,-3,2],2)
    assert Q.euler_poly(3) == Q([1,0,-6,4],4)
    assert Q.legendre_p(3) == Q([0,-3,0,5],2)
    assert Q([]).complex_roots() == []
    assert Q([1]).complex_roots() == []
    [(r,m)] = Q([1,1]).complex_roots()
    assert m == 1
    assert r.overlaps(-1)
    assert str(Q([1,2]).roots()) == "[(-1/2, 1)]"
    assert Q([2,1]).roots() == [(-2, 1)]

def test_fmpq_mat():
    Q = flint.fmpq_mat
    Z = flint.fmpz_mat
    assert Q(1,2,[3,4]) == Z(1,2,[3,4])
    assert Q(1,2,[3,4]) != Z(1,2,[5,4])
    assert Q(1,2,[3,4]) != Q(1,2,[5,4])
    assert Q(Q(1,2,[3,4])) == Q(1,2,[3,4])
    assert Q(Z(1,2,[3,4])) == Q(1,2,[3,4])
    assert Q(2,3,[1,2,3,4,5,6]) + Q(2,3,[4,5,6,7,8,9]) == Q(2,3,[5,7,9,11,13,15])
    assert Q(2,3,[1,2,3,4,5,6]) - Q(2,3,[4,5,6,7,8,9]) == Q(2,3,[-3,-3,-3,-3,-3,-3])
    assert Q(2,3,[1,2,3,4,5,6]) * Q(3,2,[4,5,6,7,8,9]) == Q(2,2,[40,46,94,109])
    assert Q(2,3,[1,2,3,4,5,6]) * Z(3,2,[4,5,6,7,8,9]) == Q(2,2,[40,46,94,109])
    assert Z(2,3,[1,2,3,4,5,6]) * Q(3,2,[4,5,6,7,8,9]) == Q(2,2,[40,46,94,109])
    assert -Q(2,1,[2,5]) == Q(2,1,[-2,-5])
    assert +Q(1,1,[3]) == Z(1,1,[3])
    assert (Q(1,1,[1]) == 1) is False
    assert (Q(1,1,[1]) != 1) is True
    assert (1 == Q(1,1,[1])) is False
    assert (1 != Q(1,1,[1])) is True
    assert Q(1,2,[3,4]) * 2 == Q(1,2,[6,8])
    assert Q(1,2,[3,4]) * flint.fmpq(1,3) == Q(1,2,[1,flint.fmpq(4,3)])
    assert Q(1,2,[3,4]) * flint.fmpq(5,3) == Q(1,2,[5,flint.fmpq(20,3)])
    assert 2 * Q(1,2,[3,4]) == Q(1,2,[6,8])
    assert flint.fmpq(1,3) * Q(1,2,[3,4]) == Q(1,2,[1,flint.fmpq(4,3)])
    M = Q([[1,2],[3,4]])
    assert M ** -1 == Q([[-4,2],[3,-1]]) / 2
    assert M ** 0 == Q([[1,0],[0,1]])
    assert M ** 1 == M
    assert M ** 2 == Q([[7,10],[15,22]])
    assert M ** 12 == Q([[138067399, 201223170],[301834755, 439902154]])
    M = Q([[1,2],[2,4]])
    assert raises(lambda: M ** -1, ZeroDivisionError)
    assert Q(1,2,[3,4]) / 2 == Q(1,2,[flint.fmpq(3,2),2])
    assert Q(1,2,[3,4]) / flint.fmpq(2,3) == Q(1,2,[flint.fmpq(9,2),6])
    assert Q(3,2,range(6)).table() == Z(3,2,range(6)).table()
    assert Q(3,2,range(6)).entries() == Z(3,2,range(6)).entries()
    assert Q(3,2,range(6)).nrows() == 3
    assert Q(3,2,range(6)).ncols() == 2
    assert Q(2,2,[3,7,4,5]).det() == -13
    assert (Q(2,2,[3,7,4,5]) / 5).det() == flint.fmpq(-13,25)
    assert raises(lambda: Q(1,2,[1,2]).det(), ValueError)
    assert Q(2,2,[1,2,3,4]).inv().inv() == Q(2,2,[1,2,3,4])
    assert raises(lambda: Q(2,2,[1,1,1,1]).inv(), ZeroDivisionError)
    assert raises(lambda: Q(2,1,[1,1]).inv(), ValueError)
    assert raises(lambda: Q([1]), TypeError)
    assert raises(lambda: Q([[1],[2,3]]), ValueError)
    assert raises(lambda: Q(None), TypeError)
    assert Q([[1,2,3],[4,5,6]]) == Q(2,3,[1,2,3,4,5,6])
    assert raises(lambda: Q(2,3,[1,2,3,4,5]), ValueError)
    assert raises(lambda: Q([[1,2,3],[4,[],6]]), TypeError)
    assert raises(lambda: Q(2,3,[1,2,3,4,[],6]), TypeError)
    assert raises(lambda: Q(2,3,[1,2],[3,4]), TypeError)
    assert bool(Q([[1]])) is True
    assert bool(Q([[0]])) is False
    assert raises(lambda: Q([[1]]) < Q([[0]]), TypeError)
    M = Q([[1,2],[3,4]])
    assert M[0,1] == 2
    M[0,1] = -1
    assert M[0,1] == -1
    # XXX: Negative indices should probably be allowed

    def set_bad(i):
        M[i,0] = -1

    raises(lambda: M[-1,0], IndexError)
    raises(lambda: M[0,-1], IndexError)
    raises(lambda: set_bad(-1), IndexError)
    raises(lambda: M[2,0], IndexError)
    raises(lambda: M[0,2], IndexError)
    raises(lambda: set_bad(2), IndexError)
    assert Q([[1,2,3],[4,5,6]]).transpose() == Q([[1,4],[2,5],[3,6]])
    raises(lambda: M + [], TypeError)
    raises(lambda: M - [], TypeError)
    raises(lambda: M * [], TypeError)
    raises(lambda: M / [], TypeError)
    raises(lambda: [] / M, TypeError)
    raises(lambda: [] + M, TypeError)
    raises(lambda: [] - M, TypeError)
    raises(lambda: [] * M, TypeError)
    # XXX: Maybe a ShapeError?
    raises(lambda: Q(1,2,[3,4]) + Q(1,3,[5,6,7]), ValueError)
    raises(lambda: Q(1,2,[3,4]) - Q(1,3,[5,6,7]), ValueError)
    raises(lambda: Q(1,2,[3,4]) * Q(1,3,[5,6,7]), ValueError)
    raises(lambda: Q(1,2,[3,4]) * Z(1,3,[5,6,7]), ValueError)
    raises(lambda: Z(1,2,[3,4]) * Q(1,3,[5,6,7]), ValueError)
    A = Q([[3,4],[5,7]]) / 11
    X = Q([[1,2],[3,4]])
    B = A*X
    assert A.solve(B) == X
    for algorithm in None, "fflu", "dixon":
        assert A.solve(B, algorithm=algorithm) == X
    assert raises(lambda: A.solve(B, algorithm="invalid"), ValueError)
    assert raises(lambda: A.solve(None), TypeError)
    assert raises(lambda: A.solve([1,2]), TypeError)
    assert raises(lambda: A.solve(Q([[1,2]])), ValueError)
    assert raises(lambda: Q([[1,2],[2,4]]).solve(Q([[1],[2]])), ZeroDivisionError)
    M = Q([[1,2,3],[flint.fmpq(1,2),5,6]])
    Mcopy = Q(M)
    Mrref = Q([[1,0,flint.fmpq(3,4)],[0,1,flint.fmpq(9,8)]])
    assert M.rref() == (Mrref, 2)
    assert M != Mrref
    assert M == Mcopy
    assert M.rref(inplace=True) == (Mrref, 2)
    assert M == Mrref
    assert M != Mcopy
    assert Q.hilbert(1, 2) == Q([[1,flint.fmpq(1,2)]])
    M2 = Q([[2, 4, 6], [8, 10, 12]]) / 4
    assert M2.numer_denom() == (Q([[1,2,3],[4,5,6]]), 2)
    half = flint.fmpq(1,2)
    M3 = Q([[half,0,0],[0,half,1],[0,0,half]])
    assert M3.charpoly() == flint.fmpq_poly([-1,6,-12,8]) / 8
    assert M3.minpoly() == flint.fmpq_poly([1,-4,4]) / 4


def test_fmpq_series():
    Qp = flint.fmpq_poly
    Q = flint.fmpq_series
    Zp = flint.fmpz_poly
    Z = flint.fmpz_series
    ctx = flint.ctx
    assert ctx.cap == 10
    s1 = Q([1,2])
    s2 = Q([1,2])
    s3 = Q([1,1])
    s4 = Q([1,2],1,11)
    s5 = Q([1,2],1,3)
    p1 = Qp([1,2])
    sz1 = Z([1,2])
    sz2 = Z([1,1])
    sz3 = Z([1,1],11)
    assert s1.prec == 10
    assert s4.prec == 11
    assert s5.prec == 3
    assert s1._equal_repr(s1) is True
    assert s1._equal_repr(s2) is True
    assert s1._equal_repr(s3) is False
    assert s1._equal_repr(s4) is False
    assert s1._equal_repr(p1) is False
    assert s1._equal_repr(sz1) is False
    assert s1._equal_repr(sz2) is False
    assert s1._equal_repr(sz3) is False
    assert Q([1])._equal_repr(flint.fmpq(1)) is False
    assert Q([1])._equal_repr(1) is False
    p = Q([1,2])
    assert p.prec == 10
    p4 = Q([1,2], prec=4)
    assert p4.prec == 4
    p4 = Q(p, prec=4)
    assert p4.prec == 4
    # XXX: this gives a core dump:
    # s = Q([1,2])
    # s[10**10] = 1
    assert Q([])._equal_repr(Q([0]))
    assert not Q([1])._equal_repr(Q([0]))
    assert Q(Q([1]))._equal_repr(Q([1]))
    assert Q(Qp([1]))._equal_repr(Q([1]))
    assert Q(Zp([1]))._equal_repr(Q([1]))
    assert Q(Z([1]))._equal_repr(Q([1]))
    assert Q(1)._equal_repr(Q([1]))
    assert Q([1],1)._equal_repr(Q([1]))
    assert Q([1],2)._equal_repr(Q([flint.fmpq(1,2)]))
    assert not Q([1],1,10)._equal_repr(Q([1],1,11))
    assert Q([1,2],3)._equal_repr(Q([flint.fmpq(1,3), flint.fmpq(2,3)]))
    assert raises(lambda: Q([1],[]), TypeError) # type: ignore
    assert raises(lambda: Q([1],0), ZeroDivisionError)
    assert raises(lambda: Q([1]) < Q([1]), TypeError) # type: ignore
    assert len(Q([1,2])) == 2
    assert Q([1,2]).length() == 2
    s5 = Q([1,2])
    assert s5[1] == 2
    assert s5[2] == 0
    assert s5[-1] == 0
    # XXX: This goes beyond cap. Should it give an error?
    assert s5[100] == 0
    s5[2] = -1
    assert s5[2] == -1
    assert s5._equal_repr(Q([1,2,-1]))

    def set_bad():
        s5[-1] = 3

    assert raises(set_bad, ValueError)
    assert Q([1,2,1]).coeffs() == list(Q([1,2,1])) == [1,2,1]
    assert Q([1,2,1],2).coeffs() == [flint.fmpq(1,2),1,flint.fmpq(1,2)]
    assert Q([1,2,0,4]).str() == "1 + 2*x + 4*x^3 + O(x^10)"
    assert Q([1,2,0,4]).repr() == "fmpq_series([1, 2, 0, 4], 1, prec=10)"
    assert Q([],1,0).str() == "O(x^0)"
    assert Q([],1,-1).str() == "(invalid power series)"
    assert (+Q([1,2]))._equal_repr(Q([1,2]))
    assert (-Q([1,2]))._equal_repr(Q([-1,-2]))
    assert (Q([1,2]) + Q([3,4,5]))._equal_repr(Q([4,6,5]))
    assert (Q([1,2]) + 1)._equal_repr(Q([2,2]))
    assert (Q([1,2]) + Qp([3,4,5]))._equal_repr(Q([4,6,5]))
    assert (1 + Q([1,2]))._equal_repr(Q([2,2]))
    assert (Qp([1,2]) + Q([3,4,5]))._equal_repr(Q([4,6,5]))
    assert raises(lambda: Q([1,2]) + [], TypeError) # type: ignore
    assert raises(lambda: [] + Q([1,2]), TypeError) # type: ignore
    assert (Q([1,2]) - Q([3,5]))._equal_repr(Q([-2,-3]))
    assert (Q([1,2]) - 1)._equal_repr(Q([0,2]))
    assert (1 - Q([1,2]))._equal_repr(Q([0,-2]))
    assert raises(lambda: [] - Q([1,2]), TypeError) # type: ignore
    assert raises(lambda: Q([1,2]) - [], TypeError) # type: ignore
    assert (Q([1,2]) * Q([1,2]))._equal_repr(Q([1,4,4]))
    assert (2 * Q([1,2]))._equal_repr(Q([2,4]))
    assert (Q([1,2]) * 2)._equal_repr(Q([2,4]))
    assert raises(lambda: [] * Q([1,2]), TypeError) # type: ignore
    assert raises(lambda: Q([1,2]) * [], TypeError) # type: ignore
    assert Q([1,2]).valuation() == 0
    assert Q([0,2]).valuation() == 1
    assert Q([0,0]).valuation() == -1
    assert (Q([1,1]) / Q([1,-1]))._equal_repr(Q([1,2,2,2,2,2,2,2,2,2]))
    assert raises(lambda: Q([1,1]) / Q([]), ZeroDivisionError)
    assert (Q([],1) / Q([1,-1]))._equal_repr(Q([],1))
    # quotient would not be a power series
    assert raises(lambda: Q([1,1]) / Q([0,1]), ValueError)
    assert (Q([1,1]) / Q([2,1]))._equal_repr(Q([512, 256, -128, 64, -32, 16, -8, 4, -2, 1], 1024))
    assert (Q([0,1,1]) / Q([0,1,-1]))._equal_repr(Q([1,2,2,2,2,2,2,2,2],1,9))
    assert (Q([2,4]) / 2)._equal_repr(Q([1,2]))
    assert (Q([1,4]) / 2)._equal_repr(Q([1,4],2))
    assert (Q([1,1]) / -1)._equal_repr(Q([-1,-1]))
    assert raises(lambda: Q([1,1]) / [], TypeError) # type: ignore
    assert raises(lambda: [] / Q([1,1]), TypeError) # type: ignore
    q = Q([1,2],3)
    assert (q ** 0)._equal_repr(Q([1]))
    assert (q ** 1)._equal_repr(q)
    assert (q ** 2)._equal_repr(Q([1,4,4],9))
    assert (q ** 3)._equal_repr(Q([1,6,12,8],27))
    assert (q ** 4)._equal_repr(Q([1, 8, 24, 32, 16], 81))
    assert (Q([1,2]) ** 2)._equal_repr(Q([1,4,4]))
    assert (Q([1,2]) ** 2)._equal_repr(Q([1,4,4]))
    assert raises(lambda: pow(Q([1,2]), 3, 5), NotImplementedError) # type: ignore
    assert (Q([1,2])(Q([0,1,2])))._equal_repr(Q([1,2,4]))
    assert raises(lambda: Q([1,2])(Q([1,2])), ValueError)
    assert raises(lambda: Q([1,2])([]), TypeError) # type: ignore
    coeffs = [0, 1, -2, 8, -40, 224, -1344, 8448, -54912, 366080]
    assert Q([0,1,2]).reversion()._equal_repr(Q(coeffs))
    # power series reversion must have valuation 1
    assert raises(lambda: Q([1,1]).reversion(), ValueError)
    assert Q([0,2,1]).reversion()._equal_repr(
        Q([0,32768,-8192,4096,-2560,1792,-1344,1056,-858,715],65536,10))
    x = Q([0,1])
    expx = x.exp()
    assert (expx)._equal_repr(Q([362880,362880,181440,60480,15120,3024,504,72,9,1],362880))
    assert (expx.inv())._equal_repr(Q([362880,-362880,181440,-60480,15120,-3024,504,-72,9,-1], 362880))
    assert (expx.derivative())._equal_repr(Q([40320,40320,20160,6720,1680,336,56,8,1],40320,prec=9))
    assert (Q([1,1]).integral())._equal_repr(Q([0,2,1],2))
    assert (expx.sqrt())._equal_repr(
        Q([185794560,92897280,23224320,3870720,483840,48384,4032,288,18,1],185794560))
    assert (expx.rsqrt())._equal_repr(
        Q([185794560,-92897280,23224320,-3870720,483840,-48384,4032,-288,18,-1],185794560))
    assert (expx.log())._equal_repr(x)
    zero = Q()
    one = Q([1])
    assert (zero.exp())._equal_repr(one)
    assert (one.log())._equal_repr(zero)
    assert (x.atan())._equal_repr(Q([0,315,0,-105,0,63,0,-45,0,35],315))
    assert (x.atanh())._equal_repr(Q([0,315,0,105,0,63,0,45,0,35],315))
    assert (x.asin())._equal_repr(Q([0,40320,0,6720,0,3024,0,1800,0,1225],40320))
    assert (x.asinh())._equal_repr(Q([0,40320,0,-6720,0,3024,0,-1800,0,1225],40320))
    assert (x.sin())._equal_repr(Q([0,362880,0,-60480,0,3024,0,-72,0,1],362880))
    assert (x.cos())._equal_repr(Q([40320,0,-20160,0,1680,0,-56,0,1],40320))
    assert (x.tan())._equal_repr(Q([0,2835,0,945,0,378,0,153,0,62],2835))
    assert (x.sinh())._equal_repr(Q([0,362880,0,60480,0,3024,0,72,0,1],362880))
    assert (x.cosh())._equal_repr(Q([40320,0,20160,0,1680,0,56,0,1],40320))
    assert (x.tanh())._equal_repr(Q([0,2835,0,-945,0,378,0,-153,0,62],2835))
    # Constant term must be nonzero
    assert raises(lambda: Q([0,1]).inv(), ValueError)
    # Constant term must be 1
    assert raises(lambda: Q([2,1]).sqrt(), ValueError)
    assert raises(lambda: Q([2,1]).rsqrt(), ValueError)
    assert raises(lambda: Q([2,1]).log(), ValueError)
    assert raises(lambda: Q([]).sqrt(), ValueError)
    assert raises(lambda: Q([]).rsqrt(), ValueError)
    assert raises(lambda: Q([]).log(), ValueError)
    # Constant term must be 0
    assert raises(lambda: Q([1,1]).exp(), ValueError)
    assert raises(lambda: Q([1,1]).atan(), ValueError)
    assert raises(lambda: Q([1,1]).atanh(), ValueError)
    assert raises(lambda: Q([1,1]).asin(), ValueError)
    assert raises(lambda: Q([1,1]).asinh(), ValueError)
    assert raises(lambda: Q([1,1]).sin(), ValueError)
    assert raises(lambda: Q([1,1]).cos(), ValueError)
    assert raises(lambda: Q([1,1]).tan(), ValueError)
    assert raises(lambda: Q([1,1]).sinh(), ValueError)
    assert raises(lambda: Q([1,1]).cosh(), ValueError)
    assert raises(lambda: Q([1,1]).tanh(), ValueError)

def test_nmod():
    G = flint.nmod
    assert G(0,2) == G(2,2) == G(-2,2)
    assert G(1,2) != G(0,2)
    assert G(0,2) != G(0,3)
    assert G(3,5) == G(8,5)
    assert G(1,2) != (1,2)
    assert isinstance(hash(G(3, 5)), int)
    assert raises(lambda: G([], 3), TypeError) # type: ignore
    #assert G(3,5) == 8        # do we want this?
    #assert 8 == G(3,5)
    assert G(3,5) != 7
    assert 7 != G(3,5)
    assert raises(lambda: G(3,5) < G(2,5), TypeError) # type: ignore
    assert bool(G(0,5)) is False
    assert bool(G(2,5)) is True
    assert G(-3,5) == -G(3,5) == G(2,5) == +G(2,5)
    assert G(2,5) + G(1,5) == G(3,5)
    assert G(2,5) + 1 == G(3,5)
    assert 1 + G(2,5) == G(3,5)
    assert G(2,5) - G(3,5) == G(4,5)
    assert G(2,5) - 3 == G(4,5)
    assert 3 - G(2,5) == G(1,5)
    assert G(2,5) * G(3,5) == G(1,5)
    assert G(2,5) * 3 == G(1,5)
    assert 3 * G(2,5) == G(1,5)
    assert G(3,17) / G(2,17) == G(10,17)
    assert G(3,17) / 2 == G(10,17)
    assert 3 / G(2,17) == G(10,17)
    assert G(0,3) / G(1,3) == G(0,3)
    assert G(3,17) * flint.fmpq(11,5) == G(10,17)
    assert G(3,17) / flint.fmpq(11,5) == G(6,17)
    assert raises(lambda: G(flint.fmpq(2, 3), 3), ZeroDivisionError)
    assert raises(lambda: G(2,5) / G(0,5), ZeroDivisionError)
    assert raises(lambda: G(2,5) / 0, ZeroDivisionError)
    assert G(1,6) / G(5,6) == G(5,6)
    assert raises(lambda: G(1,6) / G(3,6), ZeroDivisionError)
    assert G(1,3) ** 2 == G(1,3)
    assert G(2,3) ** flint.fmpz(2) == G(1,3)
    assert ~G(2,7) == G(2,7) ** -1 == G(4,7)
    assert raises(lambda: G(3,6) ** -1, ZeroDivisionError)
    assert raises(lambda: ~G(3,6), ZeroDivisionError)
    assert raises(lambda: pow(G(1,3), 2, 7), TypeError) # type: ignore
    assert G(flint.fmpq(2, 3), 5) == G(4,5)
    assert raises(lambda: G(2,5) ** G(2,5), TypeError) # type: ignore
    assert raises(lambda: flint.fmpz(2) ** G(2,5), TypeError) # type: ignore
    assert raises(lambda: G(2,5) + G(2,7), ValueError)
    assert raises(lambda: G(2,5) - G(2,7), ValueError)
    assert raises(lambda: G(2,5) * G(2,7), ValueError)
    assert raises(lambda: G(2,5) / G(2,7), ValueError)
    assert raises(lambda: G(2,5) + [], TypeError) # type: ignore
    assert raises(lambda: G(2,5) - [], TypeError) # type: ignore
    assert raises(lambda: G(2,5) * [], TypeError) # type: ignore
    assert raises(lambda: G(2,5) / [], TypeError) # type: ignore
    assert raises(lambda: G(2,5) ** [], TypeError) # type: ignore
    assert raises(lambda: [] + G(2,5), TypeError) # type: ignore
    assert raises(lambda: [] - G(2,5), TypeError) # type: ignore
    assert raises(lambda: [] * G(2,5), TypeError) # type: ignore
    assert raises(lambda: [] / G(2,5), TypeError) # type: ignore
    assert raises(lambda: [] ** G(2,5), TypeError) # type: ignore
    assert G(3,17).modulus() == 17
    assert str(G(3,5)) == "3"
    assert G(3,5).repr() == "nmod(3, 5)"

def test_nmod_poly():
    N = flint.nmod
    P = flint.nmod_poly
    Z = flint.fmpz_poly
    assert P([],17) == P([0],17)
    assert P([1,2,3],17) == P([1,2,3],17)
    assert P([1,2,3],17) != P([1,2,3],15)
    assert P([1,2,3],17) != P([1,2,4],15)
    assert P([1,2,3],17) != 1
    assert P([1,2,3],17) != Z([1,2,3])
    assert raises(lambda: P([1,2],3) < P([1,2],3), TypeError) # type: ignore
    assert P(Z([1,2,3]),17) == P([1,2,3],17)
    assert P([1,2,N(3,17)],17) == P([1,2,3],17)
    assert P(P([1,2],17),17) == P([1,2],17)
    assert raises(lambda: P(P([1,2],17),13), ValueError)
    assert raises(lambda: P([1,2,[]],17), TypeError) # type: ignore
    assert raises(lambda: P([1,2,flint.nmod(3,15)],17), ValueError)
    assert raises(lambda: P([1,2],0), ValueError)
    assert raises(lambda: P({},3), TypeError) # type: ignore
    assert P([1,2,3],17).degree() == 2
    assert P([1,2,3],17).length() == 3
    assert len(P([1,2,3],17)) == 3
    assert P([1,2,3],5).coeffs() == [N(1,5),N(2,5),N(3,5)]
    assert P([1,2,3],17) + 2 == P([3,2,3],17)
    assert 2 + P([1,2,3],17) == P([3,2,3],17)
    assert P([1,2,3],17) + P([3,4,5],17) == P([4,6,8],17)
    assert P([1,2,3],17) + P([3,4,5],17) == P([4,6,8],17)
    assert P([1,2,3],17) - 2 == P([16,2,3],17)
    assert 2 - P([1,2,3],17) == -P([16,2,3],17)
    assert +P([1,2,3],17) == P([1,2,3],17)
    assert P([1,2,3],17) - P([3,4,6],17) == P([15,15,14],17)
    assert P([1,2,3],17) * 2 == P([2,4,6],17)
    assert 2 * P([1,2,3],17) == P([2,4,6],17)
    assert P([1,2,3],17) * P([1,2,3],17) == P([1,4,10,12,9], 17)
    assert P([1,2,3],17) * Z([1,2,3]) == P([1,4,10,12,9], 17)
    assert Z([1,2,3]) * P([1,2,3],17) == P([1,4,10,12,9], 17)
    assert P([1,2,3,4,5],17) % P([2,3,4],17) == P([12,12],17)
    assert P([1,2,3,4,5],17) // P([2,3,4],17) == P([3,16,14],17)
    assert raises(lambda: P([1,2],5) // P([],5), ZeroDivisionError)
    assert raises(lambda: P([1,2],5) % P([],5), ZeroDivisionError)
    assert raises(lambda: divmod(P([1,2],5), P([],5)), ZeroDivisionError)
    assert P([1,2,3,4,5],17) ** 2 == P([1,2,3,4,5],17) * P([1,2,3,4,5],17)
    assert P([1,2,3],17) * N(3,17) == P([3,6,9],17)
    s = P([1,2,3],17)
    s2 = P([1,2,3],5)
    assert raises(lambda: s + s2, ValueError)
    assert raises(lambda: s - s2, ValueError)
    assert raises(lambda: s * s2, ValueError)
    assert raises(lambda: s // s2, ValueError)
    assert raises(lambda: s % s2, ValueError)
    assert raises(lambda: s2 + s, ValueError)
    assert raises(lambda: s2 - s, ValueError)
    assert raises(lambda: s2 * s, ValueError)
    assert raises(lambda: s2 // s, ValueError)
    assert raises(lambda: s2 % s, ValueError)
    assert raises(lambda: s + [], TypeError) # type: ignore
    assert raises(lambda: s - [], TypeError) # type: ignore
    assert raises(lambda: s * [], TypeError) # type: ignore
    assert raises(lambda: s // [], TypeError) # type: ignore
    assert raises(lambda: s % [], TypeError) # type: ignore
    assert raises(lambda: [] + s, TypeError) # type: ignore
    assert raises(lambda: [] - s, TypeError) # type: ignore
    assert raises(lambda: [] * s, TypeError) # type: ignore
    assert raises(lambda: [] // s, TypeError) # type: ignore
    assert raises(lambda: [] % s, TypeError) # type: ignore
    assert raises(lambda: [] % s, TypeError) # type: ignore
    assert raises(lambda: s.reverse(-1), ValueError)
    assert raises(lambda: s.compose("A"), TypeError) # type: ignore
    assert raises(lambda: s.compose_mod(s, "A"), TypeError) # type: ignore
    assert raises(lambda: s.compose_mod("A", P([3,6,9],17)), TypeError) # type: ignore
    assert raises(lambda: s.compose_mod(s, P([0], 17)), ZeroDivisionError)
    assert raises(lambda: pow(s, -1, P([3,6,9],17)), ValueError)
    assert raises(lambda: pow(s, 1, "A"), TypeError) # type: ignore
    assert raises(lambda: pow(s, "A", P([3,6,9],17)), TypeError) # type: ignore
    assert str(P([1,2,3],17)) == "3*x^2 + 2*x + 1"
    assert P([1,2,3],17).repr() == "nmod_poly([1, 2, 3], 17)"
    p = P([3,4,5],17)
    assert p(14) == N(2,17)
    assert p(P([1,2,3],17)) == P([12,11,11,9,11],17)
    assert raises(lambda: p({}), TypeError) # type: ignore
    p2 = P([3,4,5],17)
    assert p2[1] == N(4,17)
    assert p2[-1] == N(0,17)
    p2[1] = N(2,17)
    assert p2 == P([3,2,5],17)
    p2[1] = 6
    assert p2 == P([3,6,5],17)

    def set_bad1():
        p2[-1] = 3

    def set_bad2():
        p2[2] = [] # type: ignore

    assert raises(set_bad1, ValueError)
    assert raises(set_bad2, TypeError)
    assert bool(P([], 5)) is False
    assert bool(P([1], 5)) is True
    assert P([1,2,1],3).gcd(P([1,1],3)) == P([1,1],3)
    raises(lambda: P([1,2],3).gcd([]), TypeError) # type: ignore
    raises(lambda: P([1,2],3).gcd(P([1,2],5)), ValueError)
    p3 = P([1,2,3,4,5,6],7)
    f3 = (N(6,7), [(P([6, 1],7), 5)])
    assert p3.factor() == f3
    # XXX: factor ignores an invalid algorithm string
    for alg in [None, 'berlekamp', 'cantor-zassenhaus']:
        assert p3.factor(alg) == f3
        assert p3.factor(algorithm=alg) == f3
    assert P([1], 11).roots() == []
    assert P([1, 2, 3], 11).roots() == [(8, 1), (6, 1)]
    assert P([1, 6, 1, 8], 11).roots() == [(5, 3)]
    assert raises(lambda: P([1, 2, 3], 11).real_roots(), DomainError)
    assert raises(lambda: P([1, 2, 3], 11).complex_roots(), DomainError)


def test_nmod_mat():
    M = flint.nmod_mat
    G = flint.nmod
    Z = flint.fmpz_mat
    a = M(2,3,[1,2,3,4,5,6],17)
    b = M(2,3,[4,5,6,7,8,9],17)
    assert a == a
    assert a == M(a)
    assert a != b
    assert a.nrows() == 2
    assert a.ncols() == 3
    assert a.entries() == [G(x,17) for x in [1,2,3,4,5,6]]
    assert a.table() == [[G(x,17) for x in [1,2,3]], [G(x,17) for x in [4,5,6]]]
    assert (a + b).entries() == [G(x,17) for x in [5,7,9,11,13,15]]
    assert raises(a.det, ValueError)
    assert +a == a
    assert -a == M(2,3,[-1,-2,-3,-4,-5,-6],17)
    c = M(2,2,[1,2,3,4],17)
    assert c.det() == G(-2,17)
    assert raises(lambda: a + c, ValueError)
    assert (a * 3).entries() == [G(x,17) for x in [3,6,9,12,15,18]]
    assert (3 * a).entries() == [G(x,17) for x in [3,6,9,12,15,18]]
    assert (a * flint.fmpz(3)).entries() == [G(x,17) for x in [3,6,9,12,15,18]]
    assert (flint.fmpz(3) * a).entries() == [G(x,17) for x in [3,6,9,12,15,18]]
    assert M(2,2,[1,1,2,2],17).rank() == 1
    assert M(Z(2,2,[1,2,3,4]),17) == M(2,2,[1,2,3,4],17)
    A = M(5,3,Z.randbits(5,3,5).entries(),17)
    B = M(3,7,Z.randtest(3,7,5).entries(),17)
    C = M(7,2,Z.randtest(7,2,5).entries(),17)
    assert A*(B*C) == (A*B)*C
    assert bool(M(2,2,[0,0,0,0],17)) is False
    assert bool(M(2,2,[0,0,0,1],17)) is True
    pretty = ctx.pretty
    try:
        ctx.pretty = False
        assert repr(M(2,2,[1,2,3,4],17)) == 'nmod_mat(2, 2, [1, 2, 3, 4], 17)'
    finally:
        ctx.pretty = pretty
    assert str(M(2,2,[1,2,3,4],17)) == '[1, 2]\n[3, 4]'
    assert repr(M(2,2,[1,2,3,4],17)) == '[1, 2]\n[3, 4]'
    assert M(1,2,[3,4],17) / 3 == M(1,2,[3,4],17) * (~G(3,17))
    assert M(2,2,[1,2,3,4], 17).inv().det() == ~(M(2,2,[1,2,3,4], 17).det())
    assert M(2,2,[1,2,3,4], 17).inv().inv() == M(2,2,[1,2,3,4], 17)
    assert M(2,2,[0,1,2,3],17) * M(2, 2, [2,3,4,5], 17) == M(2,2,[4,5,16,4],17)
    assert raises(lambda: M([1], 5), TypeError)
    assert raises(lambda: M([[1],[2,3]], 5), ValueError)
    assert raises(lambda: M([[1],[2]], 0), ValueError)
    assert raises(lambda: M(None), TypeError)
    assert raises(lambda: M(None,17), TypeError)
    assert M(2,3,17) == M(2,3,[0,0,0,0,0,0],17)
    assert raises(lambda: M(2,3,[0,0,0,0,0],17), ValueError)
    assert raises(lambda: M(2,3,[0,1],[1,2],17), TypeError)
    assert M([[1,2,3],[4,5,6]], 5) == M(2,3,[1,2,3,4,5,6], 5)
    assert raises(lambda: M([[0]],13) < M([[1]],13), TypeError)
    assert (M([[1]],17) == M([[1]],13)) is False
    assert (M([[1]],17) != M([[1]],13)) is True
    assert (M([[1]],17) == None) is False
    assert (M([[1]],17) != None) is True
    M2 = M.randtest(3,4,5)
    assert all(0 <= int(x) < 5 for x in M2.entries())
    assert (M2.nrows(), M2.ncols()) == (3, 4)
    M3 = M(2,2,[1,2,3,4],17)
    assert M3[0,1] == G(2,17)
    M3_copy = M(M3)
    M3[0,1] = -1
    assert M3[0,1] == G(-1,17)
    assert M3_copy[0,1] == G(2,17)

    def set_bad(i,j):
        M3[i,j] = 2

    # XXX: negative indices should be allowed
    assert raises(lambda: M3[-1,0], IndexError)
    assert raises(lambda: M3[0,-1], IndexError)
    assert raises(lambda: set_bad(-1,0), IndexError)
    assert raises(lambda: set_bad(0,-1), IndexError)
    assert raises(lambda: M3[2,0], IndexError)
    assert raises(lambda: M3[0,2], IndexError)
    assert raises(lambda: set_bad(2,0), IndexError)
    assert raises(lambda: set_bad(0,2), IndexError)

    def set_bad2():
        M3[0,0] = 1.5

    assert raises(set_bad2, TypeError)
    assert raises(lambda: M3 + [], TypeError)
    assert raises(lambda: M3 - [], TypeError)
    assert raises(lambda: M3 * [], TypeError)
    assert raises(lambda: M3 / [], TypeError)
    assert raises(lambda: [] + M3, TypeError)
    assert raises(lambda: [] - M3, TypeError)
    assert raises(lambda: [] * M3, TypeError)
    assert raises(lambda: [] / M3, TypeError)
    assert raises(lambda: M([[1]],3) + M([[1]],5), ValueError)
    assert raises(lambda: M([[1]],3) - M([[1]],5), ValueError)
    assert raises(lambda: M([[1]],3) * M([[1]],5), ValueError)
    assert Z([[1]]) + M([[1]],3) == M([[2]],3)
    assert M([[1]],3) + Z([[1]]) == M([[2]],3)
    assert Z([[1]]) - M([[1]],3) == M([[0]],3)
    assert M([[1]],3) - Z([[1]]) == M([[0]],3)
    assert Z([[1]]) * M([[1]],3) == M([[1]],3)
    assert M([[1]],3) * Z([[1]]) == M([[1]],3)
    assert raises(lambda: M([[1]],3) - M([[1,2]],3), ValueError)
    assert raises(lambda: M([[1,2]],3) * M([[1,2]],3), ValueError)
    M4 = M([[1,2],[3,4]],17)
    assert M4.inv() == M([[15,1],[10,8]],17)
    assert raises(lambda: M([[1,2]],17).inv(), ValueError)
    assert raises(lambda: M([[1,2],[2,4]],17).inv(), ZeroDivisionError)
    assert M([[1,2,3],[4,5,6]],17).transpose() == M([[1,4],[2,5],[3,6]],17)
    M5 = M([[1,2],[3,4]],17)
    X = M([[1],[2]],17)
    b = M5*X
    assert M5.solve(b) == X
    assert raises(lambda: M5.solve([]), TypeError)
    assert raises(lambda: b.solve(M5), ValueError)
    assert raises(lambda: M([[1,2],[2,4]],17).solve(b), ZeroDivisionError)
    M6 = M([[1,2,3],[4,5,6]],17)
    M6_rref = M([[1,0,16],[0,1,2]],17)
    M6_copy = M(M6)
    assert M6.rref() == (M6_rref, 2)
    assert M6 == M6_copy
    assert M6.rref(inplace=True) == (M6_rref, 2)
    assert M6 == M6_rref
    M6 = M6_copy
    assert M6.nullspace() == (M([[1,15,1],[0,0,0],[0,0,0]],17).transpose(), 1)


def test_nmod_series():
    # XXX: currently no code in nmod_series.pyx
    pass


def test_arb():
    arb = flint.arb
    assert arb(3) > arb(2.5)
    assert arb(3) >= arb("2.5")
    assert arb(3) < arb((3,1))
    assert arb(3) <= arb("inf")
    assert arb(3) == arb(3)
    assert arb(3) != arb(2)
    assert not (arb("1.1") == arb("1.1"))

    assert arb(3).repr() == 'arb((0x3, 0x0))'
    assert arb("nan").repr() == "arb('nan')"
    assert arb(0, "inf").repr() == "arb(0.0, '+inf')"
    assert arb((1,2), (3,4)).repr() == "arb((0x1, 0x2), (0x3, 0x4))"
    assert arb(1, arb("inf")).repr() == "arb((0x1, 0x0), '+inf')"
    assert arb(1, "inf").repr() == "arb((0x1, 0x0), '+inf')"
    assert arb(1, "nan").repr() == "arb((0x1, 0x0), '+inf')"
    assert arb("nan", 1).repr() == "arb('nan')"
    assert arb("nan", "nan").repr() == "arb('nan')"

    for a in [arb(2), arb((1,2), (3,4)), arb("nan"), arb(0, "inf")]:
        assert eval(a.repr()).repr() == a.repr()


def test_acb():
    acb = flint.acb
    assert acb(1, 2).repr() == "acb(arb((0x1, 0x0)), arb((0x1, 0x1)))"


def test_pickling():
    objects = [
        flint.fmpz(1),
        flint.fmpq(1,2),
        # XXX: Add pickling for everything else
    ]
    for obj in objects:
        s = pickle.dumps(obj)
        obj2 = pickle.loads(s)
        assert obj == obj2


def test_fmpz_mod():
    from flint import fmpz_mod_ctx, fmpz, fmpz_mod

    p_sml = 163
    p_med = 2**127 - 1
    p_big = 2**255 - 19

    F_cmp = fmpz_mod_ctx(10)
    F_sml = fmpz_mod_ctx(p_sml)
    F_med = fmpz_mod_ctx(p_med)
    F_big = fmpz_mod_ctx(p_big)

    assert F_sml.is_prime() is True
    assert F_med.is_prime() is True
    assert F_big.is_prime() is True
    assert F_cmp.is_prime() is False

    # Context tests
    assert raises(lambda: fmpz_mod_ctx("AAA"), TypeError) # type: ignore
    assert raises(lambda: fmpz_mod_ctx(-1), ValueError)
    assert F_sml.modulus() == p_sml
    assert F_med.modulus() == p_med
    assert F_big.modulus() == p_big

    F_big_copy = fmpz_mod_ctx(p_big)
    assert F_big_copy == F_big
    assert F_big != F_sml
    assert hash(F_big_copy) == hash(F_big)
    assert hash(F_big) != hash(F_sml)
    assert F_big_copy != F_sml
    assert F_big_copy != "A"

    assert repr(F_sml) == "fmpz_mod_ctx(163)"
    assert str(F_sml) == "Context for fmpz_mod with modulus: 163"

    # Type tests
    assert raises(lambda: fmpz_mod(1, "AAA"), TypeError) # type: ignore

    # Test for small, medium and large char.
    for F_test in [F_sml, F_med, F_big]:
        test_mod = int(F_test.modulus())
        test_x = (-123) % test_mod # canonical value
        test_y = ((-456) % test_mod)**2 # non-canoncial value

        test_z = F_test.random_element()
        assert int(test_z) < F_test.modulus()
        assert int(test_z) >= 0

        assert raises(lambda: F_test(F_cmp(1)), ValueError)
        assert raises(lambda: F_test("abc"), NotImplementedError) # type: ignore

        F_test_copy = fmpz_mod_ctx(test_mod)
        F_other = fmpz_mod_ctx(11)

        assert raises(lambda: F_test(test_x) > 0, TypeError) # type: ignore
        assert raises(lambda: F_test(test_x) >= 0, TypeError) # type: ignore
        assert raises(lambda: F_test(test_x) < 0, TypeError) # type: ignore
        assert raises(lambda: F_test(test_x) <= 0, TypeError) # type: ignore

        assert (test_x == F_test(test_x)) is True, f"{test_x}, {F_test(test_x)}"
        assert (124 != F_test(test_x)) is True
        assert (F_test(test_x) == test_x) is True
        assert (F_test(test_x) == test_x + test_mod) is True
        assert (F_test(test_x) == 1) is False
        assert (F_test(test_x) != 1) is True
        assert (F_test(test_x) == F_test(test_x)) is True
        assert (F_test(test_x) == F_test(test_x + test_mod)) is True
        assert (F_test(test_x) == F_test(1)) is False
        assert (F_test(test_x) != F_test(1)) is True
        assert (F_test(test_x) != "abc") is True

        assert (hash(F_test(test_x)) == hash(test_x)) is True
        assert (hash(F_test(F_test(test_x))) == hash(test_x)) is True
        assert (hash(F_test(test_x)) == hash(1)) is False
        assert (hash(F_test(test_x)) != hash(1)) is True
        assert (hash(F_test(test_x)) == hash(F_test(test_x))) is True
        assert (hash(F_test(test_x)) == hash(F_test(test_x + test_mod))) is True
        assert (hash(F_test(test_x)) == hash(F_test(1))) is False
        assert (hash(F_test(test_x)) != hash(F_test(1))) is True

        # Is one, zero
        assert (F_test(0) == 0) is True
        assert F_test(0).is_zero() is True
        assert not F_test(0)
        assert not F_test(test_mod)
        assert F_test(1).is_one() is True
        assert F_test(test_mod + 1).is_one() is True
        assert F_test(1).is_one() is True
        assert F_test(2).is_one() is False

        # int, str, repr
        assert str(F_test(11)) == "11"
        assert str(F_test(-1)) == str(test_mod - 1)
        assert repr(F_test(11)) == f"fmpz_mod(11, {test_mod})"
        assert repr(F_test(-1)) == f"fmpz_mod({test_mod - 1}, {test_mod})"

        assert +F_test(5) == F_test(5)

        # Arithmetic tests

        # Negation
        assert -F_test(test_x) == F_test(-test_x) == (-test_x % test_mod)
        assert -F_test(1) == F_test(-1) == F_test(test_mod - 1)

        # Addition
        assert F_test(test_x) + F_test(test_y) == F_test(test_x + test_y)
        assert F_test(test_x) + F_test_copy(test_y) == F_test(test_x + test_y)
        assert F_test(test_x) + F_test(test_y) == F_test_copy(test_x + test_y)
        assert raises(lambda: F_test(test_x) + "AAA", TypeError) # type: ignore

        assert F_test(test_x) + F_test(test_y) == F_test(test_y) + F_test(test_x)
        assert F_test(test_x) + test_y == F_test(test_x + test_y)
        assert test_y + F_test(test_x) == F_test(test_x + test_y)
        assert F_test(test_x) + fmpz(test_y) == F_test(test_y) + F_test(test_x)
        assert raises(lambda: F_test(test_x) + F_other(test_y), ValueError)

        # Subtraction

        assert F_test(test_x) - F_test(test_y) == F_test(test_x - test_y)
        assert F_test(test_x) - test_y == F_test(test_x - test_y)
        assert F_test(test_x) - test_y == F_test(test_x) - F_test(test_y)
        assert F_test(test_y) - test_x == F_test(test_y) - F_test(test_x)
        assert test_x - F_test(test_y) == F_test(test_x) - F_test(test_y)
        assert test_y - F_test(test_x) == F_test(test_y) - F_test(test_x)
        assert F_test(test_x) - fmpz(test_y) == F_test(test_x) - F_test(test_y)
        assert raises(lambda: F_test(test_x) - F_other(test_y), ValueError)
        assert raises(lambda: F_test(test_x) - "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" - F_test(test_x), TypeError) # type: ignore

        # Multiplication

        assert F_test(test_x) * F_test(test_y) == (test_x * test_y) % test_mod
        assert F_test(test_x) * test_y == (test_x * test_y) % test_mod
        assert test_y * F_test(test_x) == (test_x * test_y) % test_mod

        assert F_test(1) * F_test(test_x) == F_test(1 * test_x)
        assert F_test(2) * F_test(test_x) == F_test(2 * test_x)
        assert F_test(3) * F_test(test_x) == F_test(3 * test_x)
        assert 1 * F_test(test_x) == F_test(1 * test_x)
        assert 2 * F_test(test_x) == F_test(2 * test_x)
        assert 3 * F_test(test_x) == F_test(3 * test_x)
        assert F_test(test_x) * 1 == F_test(1 * test_x)
        assert F_test(test_x) * 2 == F_test(2 * test_x)
        assert F_test(test_x) * 3 == F_test(3 * test_x)
        assert fmpz(1) * F_test(test_x) == F_test(1 * test_x)
        assert fmpz(2) * F_test(test_x) == F_test(2 * test_x)
        assert fmpz(3) * F_test(test_x) == F_test(3 * test_x)
        assert raises(lambda: F_test(test_x) * "AAA", TypeError) # type: ignore
        assert raises(lambda: F_test(test_x) * F_other(test_x), ValueError)

        # Exponentiation

        assert F_test(0)**0 == pow(0, 0, test_mod)
        assert F_test(0)**1 == pow(0, 1, test_mod)
        assert F_test(0)**2 == pow(0, 2, test_mod)
        assert raises(lambda: F_test(0)**(-1), ZeroDivisionError)
        assert raises(lambda: F_test(0)**("AA"), TypeError) # type: ignore

        assert F_test(test_x)**fmpz(0) == pow(test_x, 0, test_mod)
        assert F_test(test_x)**fmpz(1) == pow(test_x, 1, test_mod)
        assert F_test(test_x)**fmpz(2) == pow(test_x, 2, test_mod)
        assert F_test(test_x)**fmpz(3) == pow(test_x, 3, test_mod)

        assert F_test(test_x)**0 == pow(test_x, 0, test_mod)
        assert F_test(test_x)**1 == pow(test_x, 1, test_mod)
        assert F_test(test_x)**2 == pow(test_x, 2, test_mod)
        assert F_test(test_x)**3 == pow(test_x, 3, test_mod)
        assert F_test(test_x)**100 == pow(test_x, 100, test_mod)

        assert F_test(test_x)**(-1) == pow(test_x, -1, test_mod)
        assert F_test(test_x)**(-2) == pow(test_x, -2, test_mod)
        assert F_test(test_x)**(-3) == pow(test_x, -3, test_mod)
        assert F_test(test_x)**(-4) == pow(test_x, -4, test_mod)

        # Inversion

        assert raises(lambda: ~F_test(0), ZeroDivisionError)
        assert ~F_test(test_x) == pow(test_x, -1, test_mod)
        assert ~F_test(1) == pow(1, -1, test_mod)
        assert ~F_test(2) == pow(2, -1, test_mod), f"Broken!! {~F_test(2)}, {pow(2, -1, test_mod)}"

        assert F_test(1).inverse(check=False) == pow(1, -1, test_mod)
        assert F_test(2).inverse(check=False) == pow(2, -1, test_mod)
        assert F_test(test_x).inverse(check=False) == pow(test_x, -1, test_mod)

        # Division
        assert raises(lambda: F_test(1) / F_test(0), ZeroDivisionError)
        assert F_test(test_x) / F_test(test_y) == (test_x * pow(test_y, -1, test_mod)) % test_mod
        assert F_test(test_x) / fmpz(test_y) == (test_x * pow(test_y, -1, test_mod)) % test_mod
        assert F_test(test_x) / test_y == (test_x * pow(test_y, -1, test_mod)) % test_mod
        assert raises(lambda: F_test(test_x) / "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" / F_test(test_x), TypeError) # type: ignore
        assert raises(lambda: F_other(test_x) / F_test(test_x), ValueError)
        assert raises(lambda: F_test(test_x) // F_test(test_x), TypeError) # type: ignore
        assert 1 / F_test(2) ==  pow(2, -1, test_mod)
        assert 1 / F_test(test_x) ==  pow(test_x, -1, test_mod)
        assert 1 / F_test(test_y) ==  pow(test_y, -1, test_mod)

        assert fmpz(test_y) / F_test(test_x) == (test_y * pow(test_x, -1, test_mod)) % test_mod
        assert test_y / F_test(test_x) == (test_y * pow(test_x, -1, test_mod)) % test_mod


def test_fmpz_mod_dlog():
    from flint import fmpz, fmpz_mod_ctx

    # Input modulus must be prime
    F = fmpz_mod_ctx(4)
    g, a = F(1), F(2)
    assert raises(lambda: g.discrete_log(a), NotImplementedError)

    # Moduli must match
    F1, F2 = fmpz_mod_ctx(2), fmpz_mod_ctx(3)
    g = F1(2)
    a = F2(4)
    assert raises(lambda: g.discrete_log(a), ValueError)

    # Need to use either fmpz_mod or something which can be case to
    # fmpz
    assert raises(lambda: g.discrete_log("A"), TypeError) # type: ignore

    F = fmpz_mod_ctx(163)
    g = F(2)
    a = g**123

    assert 123 == g.discrete_log(a)

    a_int = pow(2, 123, 163)
    a_fmpz = fmpz(a_int)
    assert 123 == g.discrete_log(a_int)
    assert 123 == g.discrete_log(a_fmpz)

    # Randomised testing with smooth large modulus
    e2, e3 = 92, 79
    p = 2**e2 * 3**e3 + 1
    F = fmpz_mod_ctx(p)

    for _ in range(10):
        g = F(random.randint(1,p-1))
        for _ in range(10):
            i = random.randint(0,p)
            a = g**i
            x = g.discrete_log(a)
            assert g**x == a

def test_fmpz_mod_poly():
    from flint import fmpz_poly, fmpz_mod_poly, fmpz_mod_poly_ctx, fmpz_mod_ctx, fmpz

    # fmpz_mod_poly_ctx tests
    F = fmpz_mod_ctx(11)
    R1 = fmpz_mod_poly_ctx(F)
    R2 = fmpz_mod_poly_ctx(11)
    R3 = fmpz_mod_poly_ctx(13)

    assert raises(lambda: fmpz_mod_ctx("AAA"), TypeError) # type: ignore
    assert raises(lambda: fmpz_mod_ctx(-1), ValueError)
    assert (R1 == R1) is True
    assert (R1 != R1) is False
    assert (R1 == R2) is True
    assert (R1 != R2) is False
    assert (R1 != R3) is True
    assert (R1 == R3) is False
    assert (R1 != "AAA") is True
    assert (R1 == "AAA") is False

    assert (hash(R1) == hash(R1)) is True
    assert (hash(R1) == hash(R2)) is True
    assert (hash(R1) != hash(R3)) is True

    assert str(R1) == "Context for fmpz_mod_poly with modulus: 11"
    assert str(R1) == str(R2)
    assert repr(R3) == "fmpz_mod_poly_ctx(13)"

    assert R1.modulus() == 11

    assert R1.is_prime()
    assert R1.zero() == 0
    assert R1.one() == 1
    assert R1.gen() == R1([0,1])

    # Random testing
    f = R1.random_element()
    assert f.degree() <= 3
    f = R1.random_element(degree=5, monic=True)
    assert f.degree() == 5
    assert f.is_monic()
    f = R1.random_element(degree=100, irreducible=True)
    assert f.degree() <= 100
    assert f.is_irreducible()
    f = R1.random_element(degree=1, monic=True, irreducible=True)
    assert f.degree() == 1
    assert f.is_irreducible()
    assert f.is_monic()
    assert raises(lambda: R1.random_element(degree=-123), ValueError)
    assert raises(lambda: R1.random_element(monic="A"), TypeError) # type: ignore
    assert raises(lambda: R1.random_element(irreducible="A"), TypeError) # type: ignore

    # Conversion tests
    F = fmpz_mod_ctx(11)
    F_other = fmpz_mod_ctx(10)
    R = fmpz_mod_poly_ctx(F)
    R_other = fmpz_mod_poly_ctx(F_other)

    assert raises(lambda: fmpz_mod_poly(1, "A"), TypeError) # Need a valid context
    assert raises(lambda: R(R_other([1,2,3])), ValueError) # moduli must match
    assert raises(lambda: R(F_other(2)), ValueError) # moduli must match
    assert raises(lambda: R([F(1), F_other(2)]), ValueError) # moduli must match
    assert raises(lambda: R([F(1), "A"]), TypeError) # type: ignore

    R.__call__
    f1 = R([-1, -2, -3])
    f2 = R([fmpz(-1),fmpz(-2),fmpz(-3)])
    f3 = R([F(-1),F(-2),F(-3)])
    f4 = R(fmpz_poly([-1, -2, -3]))
    f5 = R(f4)

    assert str(f1) == "8*x^2 + 9*x + 10"
    assert str(f2) == "8*x^2 + 9*x + 10"
    assert str(f3) == "8*x^2 + 9*x + 10"
    assert str(f4) == "8*x^2 + 9*x + 10"
    assert str(f5) == "8*x^2 + 9*x + 10"

    f1 = R(5)
    f2 = R(fmpz(6))
    f3 = R(F(7))
    assert str(f1) == "5"
    assert str(f2) == "6"
    assert str(f3) == "7"

    # Printing
    f = R([5, 6, 7, 8])
    assert str(f) == "8*x^3 + 7*x^2 + 6*x + 5"
    # assert repr(f) == "fmpz_mod_poly([5, 6, 7, 8], fmpz_mod_poly_ctx(11))"

    # Get and Set tests
    f = R([5, 6, 7, 8])
    assert f[0] == 5
    assert repr(f[0]) == "fmpz_mod(5, 11)"
    f[0] = 7
    assert repr(f[0]) == "fmpz_mod(7, 11)"
    assert str(f) == "8*x^3 + 7*x^2 + 6*x + 7"

    # TODO: currently repr does pretty printing
    # just like str, we should address this. Mainly,
    # the issue is we want nice `repr` behaviour in
    # interactive shells, which currently is why this
    # choice has been made
    assert str(f) == repr(f)

    assert f[-1] == 0
    assert raises(lambda: f.__setitem__(-1, 1), ValueError)
    assert raises(lambda: f.__setitem__(1, "A"), TypeError) # type: ignore

    # Comparisons
    f1 = R([1,2,3])
    f2 = R([12,13,14])
    f3 = R([4,5,6])
    f4 = R([3])

    assert (f1 == f2) is True
    assert (f1 != f3) is True
    assert (f1 != "1") is True
    assert (f4 == 3) is True
    assert (hash(f1) == hash(f2)) is True
    assert raises(lambda: f1 > f2, TypeError) # type: ignore
    assert raises(lambda: f1 >= f2, TypeError) # type: ignore
    assert raises(lambda: f1 < f2, TypeError) # type: ignore
    assert raises(lambda: f1 <= f2, TypeError) # type: ignore

    assert len(f1) == f1.length() == 3
    assert f1.degree() == 2

    f1 = R([0])
    f2 = R([1])
    f3 = R([0, 1])

    assert f1.is_zero() is True
    assert f2.is_one() is True
    assert f3.is_gen() is True

    # Arithmetic
    p_sml = 163
    p_med = 2**127 - 1
    p_big = 2**255 - 19

    F_sml = fmpz_mod_ctx(p_sml)
    F_med = fmpz_mod_ctx(p_med)
    F_big = fmpz_mod_ctx(p_big)

    R_sml = fmpz_mod_poly_ctx(F_sml)
    R_med = fmpz_mod_poly_ctx(F_med)
    R_big = fmpz_mod_poly_ctx(F_big)

    F_cmp = fmpz_mod_ctx(10)
    R_cmp = fmpz_mod_poly_ctx(F_cmp)
    f_cmp = R_cmp([1,2,3,4,5])
    f_bad = R_cmp([2,2,2,2,2])

    for (F_test, R_test) in [(F_sml, R_sml), (F_med, R_med), (F_big, R_big)]:

        f = R_test([-1,-2])
        g = R_test([-3,-4])

        # pos, neg
        assert f is +f
        assert -f == R_test([1,2])

        # add
        assert raises(lambda: f + f_cmp, ValueError)
        assert raises(lambda: f + "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" + f, TypeError) # type: ignore
        assert f + g == R_test([-4,-6])
        assert f + 1 ==  R_test([0,-2])
        assert f + fmpz(1) ==  R_test([0,-2])
        assert f + F_test(1) ==  R_test([0,-2])
        assert 1 + f ==  R_test([0,-2])
        assert fmpz(1) + f ==  R_test([0,-2])
        assert F_test(1) + f ==  R_test([0,-2])

        # sub
        assert raises(lambda: f - f_cmp, ValueError)
        assert raises(lambda: f - "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" - f, TypeError) # type: ignore
        assert f - g == R_test([2, 2])
        assert f - 1 ==  R_test([-2,-2])
        assert f - fmpz(1) ==  R_test([-2,-2])
        assert f - F_test(1) ==  R_test([-2,-2])
        assert 1 - f ==  R_test([2, 2])
        assert fmpz(1) - f ==  R_test([2, 2])
        assert F_test(1) - f ==  R_test([2, 2])

        # mul
        assert raises(lambda: f * f_cmp, ValueError)
        assert raises(lambda: f * "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" * f, TypeError) # type: ignore
        assert f * g == R_test([3, 4 + 6, 8])
        assert f * 2 ==  R_test([-2,-4])
        assert f * fmpz(2) ==  R_test([-2,-4])
        assert f * F_test(2) ==  R_test([-2,-4])
        assert 2 * f ==  R_test([-2,-4])
        assert fmpz(2) * f ==  R_test([-2,-4])
        assert F_test(2) * f ==  R_test([-2,-4])

        # scalar_mul
        assert 2 * f == f.scalar_mul(2)
        assert raises(lambda: f.scalar_mul("AAA"), TypeError) # type: ignore

        # Exact division
        assert raises(lambda: f.exact_division(f_cmp), ValueError)
        assert raises(lambda: f.exact_division("AAA"), TypeError) # type: ignore
        assert raises(lambda: f.exact_division(0), ZeroDivisionError)

        assert (f * g).exact_division(g) == f
        assert raises(lambda: f.exact_division(g), DomainError)

        # true div
        assert raises(lambda: f / "AAA", TypeError) # type: ignore
        assert raises(lambda: f / 0, ZeroDivisionError)
        assert raises(lambda: f_cmp / 2, DomainError)

        assert (f + f) / 2 == f
        assert (f + f) / fmpz(2) == f
        assert (f + f) / F_test(2) == f

        # floor div
        assert raises(lambda: 1 // f_bad, DomainError)
        assert raises(lambda: f // f_cmp, ValueError)
        assert raises(lambda: f // "AAA", TypeError) # type: ignore
        assert raises(lambda: "AAA" // f, TypeError) # type: ignore
        assert (f * g) // g == f
        assert (f + f) // 2 == f
        assert (f + f) // fmpz(2) == f
        assert (f + f) // F_test(2) == f
        assert 2 // R_test(2) == 1
        assert (f + 1) // f == 1

        # pow
        assert raises(lambda: f**(-2), DomainError)
        assert f*f == f**2
        assert f*f == f**fmpz(2)

        # pow_mod
        # assert ui and fmpz exp agree for polynomials and generators
        R_gen = R_test.gen()
        assert pow(f, 2**60, g) == pow(pow(f, 2**30, g), 2**30, g)
        assert pow(R_gen, 2**60, g) == pow(pow(R_gen, 2**30, g), 2**30, g)

        # Check other typechecks for pow_mod
        assert raises(lambda: pow(f, -2, g), ValueError)
        assert raises(lambda: pow(f, 1, "A"), TypeError) # type: ignore
        assert raises(lambda: pow(f, "A", g), TypeError) # type: ignore
        assert raises(lambda: f.pow_mod(2**32, g, mod_rev_inv="A"), TypeError) # type: ignore

        # Shifts
        assert raises(lambda: R_test([1,2,3]).left_shift(-1), ValueError)
        assert raises(lambda: R_test([1,2,3]).right_shift(-1), ValueError)
        assert R_test([1,2,3]).left_shift(3) == R_test([0,0,0,1,2,3])
        assert R_test([1,2,3]).right_shift(1) == R_test([2,3])

        # Mod
        assert raises(lambda: f % f_bad, ValueError)
        assert raises(lambda: 123 % f_bad, DomainError)
        assert raises(lambda: f % "AAA", TypeError) # type: ignore
        assert raises(lambda: tuple() % f, TypeError) # type: ignore

        assert f % 1 == 0
        assert R_test.one() % 1 == 0
        assert 100 % R_test.one() == 0
        assert (f*g + 1) % f == 1
        assert (f*g + g) % f == (g % f)
        assert f % R_test([0,1]) == f.constant_coefficient()

        # Evaluation
        h = R_test([0, 1])
        assert h(1) == 1
        assert h(-1) == R_test.modulus() - 1
        h = R_test([0, 0, 1])
        assert h(1) == h(-1)
        assert raises(lambda: h("AAA"), TypeError) # type: ignore
        assert f([-1,-2,-3]) == [f(x) for x in [-1, -2, -3]]

        # compose
        assert raises(lambda: h.compose("AAA"), TypeError) # type: ignore

        # compose mod
        mod = R_test([1,2,3,4])
        assert f.compose(h) % mod == f.compose_mod(h, mod)
        assert raises(lambda: h.compose_mod("AAA", mod), TypeError) # type: ignore
        assert raises(lambda: h.compose_mod(f, "AAA"), TypeError) # type: ignore
        assert raises(lambda: h.compose_mod(f, R_test(0)), ZeroDivisionError)

        # Reverse
        assert raises(lambda: h.reverse(degree=-100), ValueError)
        assert R_test([-1,-2,-3]).reverse() == R_test([-3,-2,-1])

        # monic
        assert raises(lambda: f_bad.monic(), ValueError)
        assert R_test([1,2]).monic() == R_test([1 / F_test(2), 1])
        assert R_test([1,2]).monic(check=False) == R_test([1 / F_test(2), 1])

        # Square
        assert f*f == f**2 == f.square()

        # mulmod
        assert f.mul_mod(f, g) == (f*f) % g
        assert raises(lambda: f.mul_mod(f, "AAA"), TypeError) # type: ignore
        assert raises(lambda: f.mul_mod("AAA", g), TypeError) # type: ignore

        # pow_mod
        assert f.pow_mod(2, g) == (f*f) % g
        assert raises(lambda: f.pow_mod(2, "AAA"), TypeError) # type: ignore

        # divmod
        S, T = f.divmod(g)
        assert S*g + T == f
        assert raises(lambda: f.divmod("AAA"), TypeError) # type: ignore
        assert raises(lambda: f_bad.divmod(f_bad), ValueError)

        # gcd
        assert raises(lambda: f_cmp.gcd(f_cmp), DomainError)
        assert raises(lambda: f.gcd("f"), TypeError) # type: ignore

        # xgcd
        assert raises(lambda: (f_cmp).xgcd(f_cmp), ValueError)
        assert raises(lambda: (f).xgcd("f_cmp"), TypeError) # type: ignore

        # disc.
        assert raises(lambda: (f_cmp).discriminant(), NotImplementedError)

        # Radical
        assert raises(lambda: (f_cmp).radical(), NotImplementedError)

        # inverse_mod
        f_inv = f.inverse_mod(g)
        assert (f * f_inv) % g == 1
        assert raises(lambda: f.inverse_mod("AAA"), TypeError) # type: ignore
        assert raises(lambda: (f_cmp).inverse_mod(f_cmp), ValueError)

        f_inv = f.inverse_series_trunc(2)
        assert (f * f_inv) % R_test([0,0,1]) == 1
        assert raises(lambda: R_cmp([0,0,1]).inverse_series_trunc(2), ValueError)

        # Resultant
        f1 = R_test([-3, 1])
        f2 = R_test([-5, 1])
        assert f1.resultant(f2) == (3 - 5)
        assert raises(lambda: f.resultant("AAA"), TypeError) # type: ignore

        # sqrt
        f1 = R_test.random_element(irreducible=True)
        assert raises(lambda: f1.sqrt(), DomainError)
        assert (f1*f1).sqrt() in [f1, -f1]

        # deflation
        f1 = R_test([1,0,2,0,3])
        assert raises(lambda: f1.deflate(100), ValueError)
        assert f1.deflate(2) == R_test([1,2,3])

        # factor
        ff = R_test([3,2,1]) * R_test([3,2,1]) * R_test([5,4,3])
        ff_rebuild = R_test.one()
        c, facs = ff.factor()
        ff_rebuild *= c
        for p, e in facs:
            assert p.is_irreducible()
            ff_rebuild *= p**e
        assert ff_rebuild == ff

        assert set(ff.factor()[1]) == set(ff.factor(algorithm="cantor_zassenhaus")[1])
        assert set(ff.factor()[1]) == set(ff.factor(algorithm="kaltofen_shoup")[1])
        assert set(ff.factor()[1]) == set(ff.factor(algorithm="berlekamp")[1])
        assert raises(lambda: R_test([0,0,1]).factor(algorithm="AAA"), ValueError)
        assert raises(lambda: R_test([0,0,1]).real_roots(), DomainError)
        assert raises(lambda: R_test([0,0,1]).complex_roots(), DomainError)

        # composite moduli not supported
        assert raises(lambda: R_cmp([0,0,1]).factor(), DomainError)
        assert raises(lambda: R_cmp([0,0,1]).factor_squarefree(), DomainError)
        assert raises(lambda: R_cmp([0,0,1]).roots(), NotImplementedError)
        assert raises(lambda: R_cmp([0,0,1]).real_roots(), DomainError)
        assert raises(lambda: R_cmp([0,0,1]).complex_roots(), DomainError)

        # minpoly
        assert raises(lambda: R_cmp.minpoly([1,2,3,4]), NotImplementedError)
        assert raises(lambda: R_test.minpoly(1), TypeError) # type: ignore
        assert raises(lambda: R_test.minpoly([1,2,3,"AAA"]), TypeError) # type: ignore

        # multipoint_evaluation
        assert raises(lambda: R_test([1,2,3]).multipoint_evaluate([1,2,3,"AAA"]), TypeError) # type: ignore
        assert raises(lambda: R_test([1,2,3]).multipoint_evaluate("AAA"), TypeError) # type: ignore

        f = R_test([1,2,3])
        ell = [-1,-2,-3,-4,-5]
        assert [f(x) for x in ell] == f.multipoint_evaluate(ell)

        # truncate things

        f = R_test.random_element()
        g = R_test.random_element()
        h = R_test.random_element()
        x = R_test.gen()

        f_trunc = f % x**3

        assert f.equal_trunc(f_trunc, 3)
        assert not f.equal_trunc("A", 3)
        assert not f.equal_trunc(f_cmp, 3)

        assert raises(lambda: f.add_trunc("A", 1), TypeError) # type: ignore
        assert raises(lambda: f.add_trunc(f_cmp, 1), ValueError)
        assert f.add_trunc(g, 3) == (f + g) % x**3

        assert raises(lambda: f.sub_trunc("A", 1), TypeError) # type: ignore
        assert raises(lambda: f.sub_trunc(f_cmp, 1), ValueError)
        assert f.sub_trunc(g, 3) == (f - g) % x**3

        assert raises(lambda: f.mul_low("A", h), TypeError) # type: ignore
        assert raises(lambda: f.mul_low(g, "A"), TypeError) # type: ignore
        assert f.mul_low(g, 3) == (f * g) % x**3

        assert raises(lambda: f.pow_trunc(-1, 5), ValueError)


def test_fmpz_mod_mat():
    c11 = flint.fmpz_mod_ctx(11)
    c13 = flint.fmpz_mod_ctx(13)

    M11_1 = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c11)
    M11_2 = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c11)
    M13 = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c13)
    assert (M11_1 == M11_2) is True
    assert (M11_1 != M11_2) is False
    assert (M11_1 == M13) is False
    assert (M11_1 != M13) is True

    Mn_11 = flint.nmod_mat([[1,2,3],[4,5,6],[7,8,9]], 11)
    Mn_13 = flint.nmod_mat([[1,2,3],[4,5,6],[7,8,9]], 13)
    assert (M11_1 == Mn_11) is True
    assert (M11_1 != Mn_11) is False
    assert (M11_1 == Mn_13) is False
    assert (M11_1 != Mn_13) is True
    assert (Mn_11 == M11_1) is True
    assert (Mn_11 != M11_1) is False
    assert (Mn_13 == M11_1) is False
    assert (Mn_13 != M11_1) is True

    M = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c11)
    assert M.nrows() == 3
    assert M.ncols() == 3
    assert M.entries() == [flint.fmpz_mod(i, c11) for i in [1,2,3,4,5,6,7,8,9]]
    assert isinstance(M[0,0], flint.fmpz_mod)
    assert M[0,0] == 1
    assert raises(lambda: flint.fmpz_mod_mat([[1]]), TypeError)
    assert raises(lambda: flint.fmpz_mod_mat([[1,2],[3,4]], c11, c11), TypeError)
    assert raises(lambda: flint.fmpz_mod_mat([[1,2],[3,4]], None), TypeError)

    M11 = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c11)
    M13 = flint.fmpz_mod_mat([[1,2,3],[4,5,6],[7,8,9]], c13)
    Mz = flint.fmpz_mat([[1,2,3],[4,5,6],[7,8,9]])
    Mn_11 = flint.nmod_mat([[1,2,3],[4,5,6],[7,8,9]], 11)
    Mn_13 = flint.nmod_mat([[1,2,3],[4,5,6],[7,8,9]], 13)
    assert raises(lambda: flint.fmpz_mod_mat(Mz), TypeError)
    assert flint.fmpz_mod_mat(Mz, c11) == M11
    assert flint.fmpz_mod_mat(Mz, c13) == M13
    assert flint.fmpz_mod_mat(Mn_11) == M11
    assert flint.fmpz_mod_mat(Mn_13) == M13
    assert flint.fmpz_mod_mat(Mn_11, c11) == M11
    assert flint.fmpz_mod_mat(Mn_13, c13) == M13
    assert raises(lambda: flint.fmpz_mod_mat(Mn_11, c13), TypeError)
    assert raises(lambda: flint.fmpz_mod_mat(Mn_13, c11), TypeError)
    assert flint.fmpz_mod_mat(M11) == M11
    assert flint.fmpz_mod_mat(M13) == M13
    assert flint.fmpz_mod_mat(M11, c11) == M11
    assert flint.fmpz_mod_mat(M13, c13) == M13
    assert raises(lambda: flint.fmpz_mod_mat(M11, c13), TypeError)
    assert raises(lambda: flint.fmpz_mod_mat(M13, c11), TypeError)

    A = flint.arb_mat([[1,2,3],[4,5,6],[7,8,9]])
    assert raises(lambda: flint.fmpz_mod_mat(A, c11), TypeError)


def test_division_scalar():
    Z = flint.fmpz
    Q = flint.fmpq
    F17 = lambda x: flint.nmod(x, 17)
    ctx = flint.fmpz_mod_ctx(163)
    F163 = lambda a: flint.fmpz_mod(a, ctx)
    # fmpz exact division
    for (a, b) in [(Z(4), Z(2)), (Z(4), 2), (4, Z(2))]:
        assert a / b == Z(2)
    for (a, b) in [(Z(5), Z(2)), (Z(5), 2), (5, Z(2))]:
        assert raises(lambda: a / b, DomainError)
    # fmpz Euclidean division
    for (a, b) in [(Z(5), Z(2)), (Z(5), 2), (5, Z(2))]:
        assert a // b == 2
        assert a % b == 1
        assert divmod(a, b) == (2, 1)
    # field division
    for (a, b) in [(Q(5), Q(2)), (Q(5), 2), (5, Q(2))]:
        assert a / b == Q(5,2)
    for (a, b) in [(F17(5), F17(2)), (F17(5), 2), (5, F17(2))]:
        assert a / b == F17(11)
    for (a, b) in [(F163(5), F163(2)), (F163(5), 2), (5, F163(2))]:
        assert a / b == F163(84)
    # divmod with fields - should this give remainder zero instead of error?
    for K in [Q, F17, F163]:
        for (a, b) in [(K(5), K(2)), (K(5), 2), (5, K(2))]:
            assert raises(lambda: divmod(a, b), TypeError)
    # Zero division
    for R in [Z, Q, F17, F163]:
        assert raises(lambda: R(5) / 0, ZeroDivisionError)
        assert raises(lambda: R(5) / R(0), ZeroDivisionError)
        assert raises(lambda: 5 / R(0), ZeroDivisionError)
    # Bad types
    for R in [Z, Q, F17, F163]:
        assert raises(lambda: R(5) / "AAA", TypeError)
        assert raises(lambda: "AAA" / R(5), TypeError)


def test_division_poly():
    Z = flint.fmpz
    Q = flint.fmpq
    F17 = lambda x: flint.nmod(x, 17)
    ctx = flint.fmpz_mod_ctx(163)
    F163 = lambda a: flint.fmpz_mod(a, ctx)
    PZ = lambda x: flint.fmpz_poly(x)
    PQ = lambda x: flint.fmpq_poly(x)
    PF17 = lambda x: flint.nmod_poly(x, 17)
    PF163 = lambda x: flint.fmpz_mod_poly(x, flint.fmpz_mod_poly_ctx(163))
    # fmpz exact scalar division
    assert PZ([2, 4]) / Z(2) == PZ([1, 2])
    assert PZ([2, 4]) / 2 == PZ([1, 2])
    assert raises(lambda: PZ([2, 5]) / Z(2), DomainError)
    assert raises(lambda: PZ([2, 5]) / 2, DomainError)
    # field division by scalar
    for (K, PK) in [(Q, PQ), (F17, PF17), (F163, PF163)]:
        assert PK([2, 5]) / K(2) == PK([K(2)/K(2), K(5)/K(2)])
        assert PK([2, 5]) / 2 == PK([K(2)/K(2), K(5)/K(2)])
    # No other scalar division is allowed
    for (R, PR) in [(Z, PZ), (Q, PQ), (F17, PF17), (F163, PF163)]:
        assert raises(lambda: R(2) / PR([2, 5]), DomainError)
        assert raises(lambda: 2 / PR([2, 5]), DomainError)
        assert raises(lambda: PR([2, 5]) / 0, ZeroDivisionError)
        assert raises(lambda: PR([2, 5]) / R(0), ZeroDivisionError)
    # exact polynomial division
    for (R, PR) in [(Z, PZ), (Q, PQ), (F17, PF17), (F163, PF163)]:
        assert PR([2, 4]) / PR([1, 2]) == PR([2])
        assert PR([2, -3, 1]) / PR([-1, 1]) == PR([-2, 1])
        assert raises(lambda: PR([2, 4]) / PR([1, 3]), DomainError)
        assert PR([2]) / PR([2]) == 2 / PR([2]) == PR([1])
        assert PR([0]) / PR([1, 2]) == 0 / PR([1, 2]) == PR([0])
        if R is Z:
            assert raises(lambda: PR([1, 2]) / PR([2, 4]), DomainError)
            assert raises(lambda: 1 / PR([2]), DomainError)
        else:
            assert PR([1, 2]) / PR([2, 4]) == PR([R(1)/R(2)])
            assert 1 / PR([2]) == PR([R(1)/R(2)])
        assert raises(lambda: PR([1, 2]) / PR([0]), ZeroDivisionError)
    # Euclidean polynomial division
    for (R, PR) in [(Z, PZ), (Q, PQ), (F17, PF17), (F163, PF163)]:
        assert PR([2, 4]) // PR([1, 2]) == PR([2])
        assert PR([2, 4]) % PR([1, 2]) == PR([0])
        assert divmod(PR([2, 4]), PR([1, 2])) == (PR([2]), PR([0]))
        assert PR([3, -3, 1]) // PR([-1, 1]) == PR([-2, 1])
        assert PR([3, -3, 1]) % PR([-1, 1]) == PR([1])
        assert divmod(PR([3, -3, 1]), PR([-1, 1])) == (PR([-2, 1]), PR([1]))
        assert PR([2]) // PR([2]) == 2 // PR([2]) == PR([1])
        assert PR([2]) % PR([2]) == 2 % PR([2]) == PR([0])
        assert divmod(PR([2]), PR([2])) == (PR([1]), PR([0]))
        assert PR([0]) // PR([1, 2]) == 0 // PR([1, 2]) == PR([0])
        assert PR([0]) % PR([1, 2]) == 0 % PR([1, 2]) == PR([0])
        assert divmod(PR([0]), PR([1, 2])) == (PR([0]), PR([0]))
        if R is Z:
            assert PR([2, 2]) // PR([2, 4]) == PR([2, 2]) // PR([2, 4]) == PR([0])
            assert PR([2, 2]) % PR([2, 4]) == PR([2, 2]) % PR([2, 4]) == PR([2, 2])
            assert divmod(PR([2, 2]), PR([2, 4])) == (PR([0]), PR([2, 2]))
            assert 1 // PR([2]) == PR([1]) // PR([2]) == PR([0])
            assert 1 % PR([2]) == PR([1]) % PR([2]) == PR([1])
            assert divmod(1, PR([2])) == (PR([0]), PR([1]))
        else:
            assert PR([2, 2]) // PR([2, 4]) == PR([R(1)/R(2)])
            assert PR([2, 2]) % PR([2, 4]) == PR([1])
            assert divmod(PR([2, 2]), PR([2, 4])) == (PR([R(1)/R(2)]), PR([1]))
            assert 1 // PR([2]) == PR([R(1)/R(2)])
            assert 1 % PR([2]) == PR([0])
            assert divmod(1, PR([2])) == (PR([R(1)/R(2)]), PR([0]))


def test_division_matrix():
    Z = flint.fmpz
    Q = flint.fmpq
    F17 = lambda x: flint.nmod(x, 17)
    ctx = flint.fmpz_mod_ctx(163)
    F163 = lambda a: flint.fmpz_mod(a, ctx)
    MZ = lambda x: flint.fmpz_mat(x)
    MQ = lambda x: flint.fmpq_mat(x)
    MF17 = lambda x: flint.nmod_mat(x, 17)
    MF163 = lambda x: flint.fmpz_mod_mat(x, ctx)
    # fmpz exact division
    assert MZ([[2, 4]]) / Z(2) == MZ([[1, 2]])
    assert MZ([[2, 4]]) / 2 == MZ([[1, 2]])
    assert raises(lambda: MZ([[2, 5]]) / Z(2), DomainError)
    assert raises(lambda: MZ([[2, 5]]) / 2, DomainError)
    # field division by scalar
    for (K, MK) in [(Q, MQ), (F17, MF17), (F163, MF163)]:
        assert MK([[2, 5]]) / K(2) == MK([[K(2)/K(2), K(5)/K(2)]])
        assert MK([[2, 5]]) / 2 == MK([[K(2)/K(2), K(5)/K(2)]])
    # No other division is allowed
    for (R, MR) in [(Z, MZ), (Q, MQ), (F17, MF17), (F163, MF163)]:
        M = MR([[2, 5]])
        for s in (2, R(2)):
            assert raises(lambda: s / M, TypeError)
            assert raises(lambda: M // s, TypeError)
            assert raises(lambda: s // M, TypeError)
            assert raises(lambda: M % s, TypeError)
            assert raises(lambda: s % M, TypeError)
            assert raises(lambda: divmod(s, M), TypeError)
            assert raises(lambda: divmod(M, s), TypeError)
        assert raises(lambda: M / M, TypeError)
        assert raises(lambda: M // M, TypeError)
        assert raises(lambda: M % M, TypeError)
        assert raises(lambda: divmod(M, M), TypeError)
        assert raises(lambda: M / 0, ZeroDivisionError)
        assert raises(lambda: M / R(0), ZeroDivisionError)


def _all_polys() -> list[tuple[Any, Any, bool, flint.fmpz]]:
    return [
        # (poly_type, scalar_type, is_field, characteristic)

        # Z and Q
        (flint.fmpz_poly, flint.fmpz, False, flint.fmpz(0)),
        (flint.fmpq_poly, flint.fmpq, True, flint.fmpz(0)),

        # Z/pZ (p prime)
        (lambda *a: flint.nmod_poly(*a, 17), lambda x: flint.nmod(x, 17), True, flint.fmpz(17)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(163)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(163)),
         True, flint.fmpz(163)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(2**127 - 1)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(2**127 - 1)),
         True, flint.fmpz(2**127 - 1)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(2**255 - 19)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(2**255 - 19)),
         True, flint.fmpz(2**255 - 19)),

        # GF(p^k) (p prime)
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(2**127 - 1)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(2**127 - 1)),
         True, flint.fmpz(2**127 - 1)),
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(2**127 - 1, 2)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(2**127 - 1, 2)),
         True, flint.fmpz(2**127 - 1)),
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(65537)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(65537)),
         True, flint.fmpz(65537)),
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(65537, 5)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(65537, 5)),
         True, flint.fmpz(65537)),
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(11)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(11)),
         True, flint.fmpz(11)),
        (lambda *a: flint.fq_default_poly(*a, flint.fq_default_poly_ctx(11, 5)),
         lambda x: flint.fq_default(x, flint.fq_default_ctx(11, 5)),
         True, flint.fmpz(11)),

        # Z/nZ (n composite)
        (lambda *a: flint.nmod_poly(*a, 16), lambda x: flint.nmod(x, 16), False, flint.fmpz(16)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(164)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(164)),
         False, flint.fmpz(164)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(2**127)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(2**127)),
         False, flint.fmpz(2**127)),
        (lambda *a: flint.fmpz_mod_poly(*a, flint.fmpz_mod_poly_ctx(2**255)),
         lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(2**255)),
         False, flint.fmpz(2**255)),
    ]


Tpoly = TypeVar("Tpoly", bound=typ.epoly_p)
Tc = TypeVar("Tc", bound=typ.scalar_p)
TS = Callable[[Tc | int], Tc]
TP = Callable[[Tpoly | Sequence[Tc | int] | Tc | int], Tpoly]
_PolyTestCase = tuple[TP[Tpoly,Tc], TS[Tc], bool, flint.fmpz]


def _for_all_polys(test: Callable[[_PolyTestCase], None]) -> None:
    """Test all mpoly types with the given test function."""
    # Spell it out like this so that a type checker can understand the types
    # in the generics for each call of test().

    fmpz: TS[flint.fmpz] = flint.fmpz
    fmpq: TS[flint.fmpq] = flint.fmpq
    fmpz_poly: TP[flint.fmpz_poly, flint.fmpz] = flint.fmpz_poly
    fmpq_poly: TP[flint.fmpq_poly, flint.fmpq] = flint.fmpq_poly

    def nmod_poly(
        p: int,
    ) -> tuple[TP[flint.nmod_poly, flint.nmod], TS[flint.nmod]]:
        """Make nmod poly and scalar constructors for modulus p."""

        def poly(
            x: Sequence[flint.nmod | int] | flint.nmod_poly | flint.nmod | int, /
        ) -> flint.nmod_poly:
            return flint.nmod_poly(x, p)

        def elem(x: int | flint.nmod = 0, /) -> flint.nmod:
            return flint.nmod(x, p)

        return poly, elem

    def fmpz_mod_poly(
        p: int,
    ) -> tuple[TP[flint.fmpz_mod_poly, flint.fmpz_mod], TS[flint.fmpz_mod]]:
        """Make fmpz_mod poly and scalar constructors for modulus p."""
        ectx = flint.fmpz_mod_ctx(p)
        pctx = flint.fmpz_mod_poly_ctx(ectx)

        def poly(
            x: Sequence[flint.fmpz_mod | int]
            | flint.fmpz_mod_poly
            | flint.fmpz_mod
            | int,
            /,
        ) -> flint.fmpz_mod_poly:
            return flint.fmpz_mod_poly(x, pctx)

        def elem(x: int | flint.fmpz_mod = 0, /) -> flint.fmpz_mod:
            return flint.fmpz_mod(x, ectx)

        return poly, elem

    def fq_default_poly(
        p: int, k: int | None = None
    ) -> tuple[
        TP[flint.fq_default_poly, flint.fq_default], TS[flint.fq_default]
    ]:
        """Make fq_default poly and scalar constructors for field p^k."""
        if k is None:
            ectx = flint.fq_default_ctx(p)
        else:
            ectx = flint.fq_default_ctx(p, k)
        pctx = flint.fq_default_poly_ctx(ectx)

        def poly(
            x: Sequence[flint.fq_default | int]
            | flint.fq_default_poly
            | flint.fq_default
            | int,
            /,
        ) -> flint.fq_default_poly:
            return flint.fq_default_poly(x, pctx)

        def elem(x: int | flint.fq_default = 0, /) -> flint.fq_default:
            return flint.fq_default(x, ectx)

        return poly, elem

    # Z and Q
    test((fmpz_poly, fmpz, False, flint.fmpz(0)))
    test((fmpq_poly, fmpq, True, flint.fmpz(0)))

    # Z/pZ (p prime)
    test((*nmod_poly(17), True, flint.fmpz(17)))
    test((*fmpz_mod_poly(163), True, flint.fmpz(163)))
    test((*fmpz_mod_poly(2**127 - 1), True, flint.fmpz(2**127 - 1)))
    test((*fmpz_mod_poly(2**255 - 19), True, flint.fmpz(2**255 - 19)))

    # Z/nZ (n composite)
    test((*nmod_poly(16), False, flint.fmpz(16)))
    test((*fmpz_mod_poly(164), False, flint.fmpz(164)))
    test((*fmpz_mod_poly(2**127), False, flint.fmpz(2**127)))
    test((*fmpz_mod_poly(2**255), False, flint.fmpz(2**255)))

    # GF(p^k)
    test((*fq_default_poly(2**127 - 1), True, flint.fmpz(2**127 - 1)))
    test((*fq_default_poly(2**127 - 1, 2), True, flint.fmpz(2**127 - 1)))
    test((*fq_default_poly(65537), True, flint.fmpz(65537)))
    test((*fq_default_poly(65537, 5), True, flint.fmpz(65537)))
    test((*fq_default_poly(11), True, flint.fmpz(11)))
    test((*fq_default_poly(11, 5), True, flint.fmpz(11)))


def all_polys(f: Callable[[_PolyTestCase], None]) -> Callable[[], None]:
    """Decorator to run a test function for all mpoly types."""
    def wrapper():
        _for_all_polys(f)
    wrapper.__name__ = f.__name__
    return wrapper


@all_polys
def test_polys(args: _PolyTestCase[typ.epoly_p[Tc], Tc]) -> None:
    # To test type annotations, uncomment:
    # P: type[flint.fmpq_poly]
    # S: type[flint.fmpq]
    # is_field: bool
    # characteristic: flint.fmpz
    P, S, is_field, characteristic = args

    composite_characteristic = characteristic != 0 and not characteristic.is_prime()
    # nmod_poly crashes for many operations with non-prime modulus
    #     https://github.com/flintlib/python-flint/issues/124
    # so we can't even test it...
    nmod_poly_will_crash = type(P(1)) is flint.nmod_poly and composite_characteristic

    assert P([S(1)]) == P([1]) == P(P([1])) == P(1)

    assert raises(lambda: P([None]), TypeError) # type: ignore
    assert raises(lambda: P(object()), TypeError) # type: ignore
    assert raises(lambda: P(None), TypeError) # type: ignore
    assert raises(lambda: P(None, None), TypeError) # type: ignore
    assert raises(lambda: P([1,2], None), TypeError) # type: ignore
    assert raises(lambda: P(1, None), TypeError) # type: ignore

    assert len(P([])) == P([]).length() == 0
    assert len(P([1])) == P([1]).length() == 1
    assert len(P([1,2])) == P([1,2]).length() == 2
    assert len(P([1,2,3])) == P([1,2,3]).length() == 3

    assert P([]).degree() == -1
    assert P([1]).degree() == 0
    assert P([1,2]).degree() == 1
    assert P([1,2,3]).degree() == 2

    assert (P([1]) == P([1])) is True
    assert (P([1]) != P([1])) is False
    assert (P([1]) == P([2])) is False
    assert (P([1]) != P([2])) is True

    assert (P([1]) == 1) is True
    assert (P([1]) != 1) is False
    assert (P([1]) == 2) is False
    assert (P([1]) != 2) is True

    assert (1 == P([1])) is True
    assert (1 != P([1])) is False
    assert (2 == P([1])) is False
    assert (2 != P([1])) is True

    s1, s2 = S(1), S(2)

    assert (P([s1]) == s1) is True
    assert (P([s1]) != s1) is False
    assert (P([s1]) == s2) is False
    assert (P([s1]) != s2) is True

    assert (s1 == P([s1])) is True
    assert (s1 != P([s1])) is False
    assert (s1 == P([s2])) is False
    assert (s1 != P([s2])) is True

    assert (P([1]) == None) is False
    assert (P([1]) != None) is True
    assert (None == P([1])) is False
    assert (None != P([1])) is True

    assert raises(lambda: P([1]) < P([1]), TypeError) # type: ignore
    assert raises(lambda: P([1]) <= P([1]), TypeError) # type: ignore
    assert raises(lambda: P([1]) > P([1]), TypeError) # type: ignore
    assert raises(lambda: P([1]) >= P([1]), TypeError) # type: ignore
    assert raises(lambda: P([1]) < None, TypeError) # type: ignore
    assert raises(lambda: P([1]) <= None, TypeError) # type: ignore
    assert raises(lambda: P([1]) > None, TypeError) # type: ignore
    assert raises(lambda: P([1]) >= None, TypeError) # type: ignore
    assert raises(lambda: None < P([1]), TypeError) # type: ignore
    assert raises(lambda: None <= P([1]), TypeError) # type: ignore
    assert raises(lambda: None > P([1]), TypeError) # type: ignore
    assert raises(lambda: None >= P([1]), TypeError) # type: ignore

    assert P([1, 2, 3])[1] == S(2)
    assert P([1, 2, 3])[-1] == S(0)
    assert P([1, 2, 3])[3] == S(0)

    p = P([1, 2, 3])
    p[1] = S(4)
    assert p == P([1, 4, 3])

    def setbad(obj, i, val):
        obj[i] = val

    assert raises(lambda: setbad(p, 2, None), TypeError)
    assert raises(lambda: setbad(p, -1, 1), ValueError)

    for v in [], [1], [1, 2]:
        p = P(v)
        if type(p) == flint.fmpz_poly:
            assert p.repr() == f'fmpz_poly({v!r})'
        elif type(p) == flint.fmpq_poly:
            assert p.repr() == f'fmpq_poly({v!r})'
        elif type(p) == flint.nmod_poly:
            assert p.repr() == f'nmod_poly({v!r}, {p.modulus()})'
        elif type(p) == flint.fmpz_mod_poly:
            assert p.repr() == f'fmpz_mod_poly({v!r}, {p.context()!r})'
        elif type(p) == flint.fq_default_poly:
            assert p.repr() == f'fq_default_poly({v!r}, {p.context()!r})'
        else:
            assert False

    assert repr(P([])) == '0'
    assert repr(P([1])) == '1'
    assert repr(P([1, 2])) == '2*x + 1'
    assert repr(P([1, 2, 3])) == '3*x^2 + 2*x + 1'

    p = P([1, 2, 3])
    assert p(0) == p(S(0)) == S(1) == 1
    assert p(1) == p(S(1)) == S(6) == 6
    assert p(p) == P([6, 16, 36, 36, 27])
    assert raises(lambda: p(None), TypeError) # type: ignore

    assert bool(P([])) is False
    assert bool(P([1])) is True

    assert P([]).is_zero() is True
    assert P([1]).is_zero() is False

    assert P([]).is_one() is False
    assert P([1]).is_one() is True

    assert +P([1, 2, 3]) == P([1, 2, 3])
    assert -P([1, 2, 3]) == P([-1, -2, -3])

    assert P([1, 2, 3]) + P([4, 5, 6]) == P([5, 7, 9])

    for T in (int, S, flint.fmpz):
        assert P([1, 2, 3]) + T(1) == P([2, 2, 3])
        assert T(1) + P([1, 2, 3]) == P([2, 2, 3])

    assert raises(lambda: P([1, 2, 3]) + None, TypeError) # type: ignore
    assert raises(lambda: None + P([1, 2, 3]), TypeError) # type: ignore

    assert P([1, 2, 3]) - P([4, 5, 6]) == P([-3, -3, -3])

    for T in (int, S, flint.fmpz):
        assert P([1, 2, 3]) - T(1) == P([0, 2, 3])
        assert T(1) - P([1, 2, 3]) == P([0, -2, -3])

    assert raises(lambda: P([1, 2, 3]) - None, TypeError) # type: ignore
    assert raises(lambda: None - P([1, 2, 3]), TypeError) # type: ignore

    assert P([1, 2, 3]) * P([4, 5, 6]) == P([4, 13, 28, 27, 18])

    for T in (int, S, flint.fmpz):
        assert P([1, 2, 3]) * T(2) == P([2, 4, 6])
        assert T(2) * P([1, 2, 3]) == P([2, 4, 6])

    assert raises(lambda: P([1, 2, 3]) * None, TypeError) # type: ignore
    assert raises(lambda: None * P([1, 2, 3]), TypeError) # type: ignore

    assert P([1, 2, 1]) // P([1, 1]) == P([1, 1])
    assert P([1, 2, 1]) % P([1, 1]) == P([0])
    assert divmod(P([1, 2, 1]), P([1, 1])) == (P([1, 1]), P([0]))

    if is_field:
        assert P([1, 1]) // 2 == P([S(1)/2, S(1)/2])
        assert P([1, 1]) % 2 == P([0])
    elif characteristic == 0:
        assert P([1, 1]) // 2 == P([0, 0])
        assert P([1, 1]) % 2 == P([1, 1])
    elif nmod_poly_will_crash:
        pass
    else:
        # Z/nZ for n not prime
        if characteristic % 2 == 0:
            assert raises(lambda: P([1, 1]) // 2, DomainError)
            assert raises(lambda: P([1, 1]) % 2, DomainError)
        else:
            assert False

    assert 1 // P([1, 1]) == P([0])
    assert 1 % P([1, 1]) == P([1])
    assert divmod(1, P([1, 1])) == (P([0]), P([1]))

    assert raises(lambda: P([1, 2, 1]) // None, TypeError) # type: ignore
    assert raises(lambda: P([1, 2, 1]) % None, TypeError) # type: ignore
    assert raises(lambda: divmod(P([1, 2, 1]), None), TypeError) # type: ignore

    assert raises(lambda: None // P([1, 1]), TypeError) # type: ignore
    assert raises(lambda: None % P([1, 1]), TypeError) # type: ignore
    assert raises(lambda: divmod(None, P([1, 1])), TypeError) # type: ignore

    assert raises(lambda: P([1, 2, 1]) // 0, ZeroDivisionError)
    assert raises(lambda: P([1, 2, 1]) % 0, ZeroDivisionError)
    assert raises(lambda: divmod(P([1, 2, 1]), 0), ZeroDivisionError)

    assert raises(lambda: P([1, 2, 1]) // P([0]), ZeroDivisionError)
    assert raises(lambda: P([1, 2, 1]) % P([0]), ZeroDivisionError)
    assert raises(lambda: divmod(P([1, 2, 1]), P([0])), ZeroDivisionError)

    # Exact/field scalar division
    if is_field:
        assert P([2, 2]) / 2 == P([1, 1])
        assert P([1, 2]) / 2 == P([S(1)/2, 1])
    elif characteristic == 0:
        assert P([2, 2]) / 2 == P([1, 1])
        assert raises(lambda: P([1, 2]) / 2, DomainError)
    elif nmod_poly_will_crash:
        pass
    else:
        # Z/nZ for n not prime
        assert raises(lambda: P([2, 2]) / 2, DomainError)
        assert raises(lambda: P([1, 2]) / 2, DomainError)

    assert raises(lambda: P([1, 2]) / 0, ZeroDivisionError)

    if not composite_characteristic:
        assert P([1, 2, 1]) / P([1, 1]) == P([1, 1])
        assert raises(lambda: 1 / P([1, 1]), DomainError)
        assert raises(lambda: P([1, 2, 1]) / P([1, 2]), DomainError)

    # Shifts and truncation.
    assert P([1, 2, 3]).left_shift(3) == P([0, 0, 0, 1, 2, 3])
    assert P([1, 2, 3]).right_shift(1) == P([2, 3])
    assert P([1, 2, 3]).truncate(4) == P([1, 2, 3])
    assert P([1, 2, 3]).truncate(2) == P([1, 2])

    assert P([1, 1]) ** 0 == P([1])
    assert P([1, 1]) ** 1 == P([1, 1])
    assert P([1, 1]) ** 2 == P([1, 2, 1])
    assert raises(lambda: P([1, 1]) ** -1, DomainError)
    assert raises(lambda: P([1, 1]) ** None, TypeError) # type: ignore

    # XXX: Not sure what this should do in general:
    p = P([1, 1])
    mod = P([1, 1])
    if isinstance(p, flint.fmpz_mod_poly) and isinstance(mod, flint.fmpz_mod_poly):
        assert p * p % mod == pow(p, 2, mod)
    elif isinstance(p, flint.nmod_poly) and isinstance(mod, flint.nmod_poly):
        assert p * p % mod == pow(p, 2, mod)
    elif isinstance(p, flint.fq_default_poly) and isinstance(mod, flint.fq_default_poly):
        assert p * p % mod == pow(p, 2, mod)
    else:
        assert raises(lambda: pow(p, 2, mod), NotImplementedError) # type: ignore

    if not composite_characteristic:
        assert P([1, 2, 1]).gcd(P([1, 1])) == P([1, 1])
        assert raises(lambda: P([1, 2, 1]).gcd(None), TypeError) # type: ignore
    elif nmod_poly_will_crash:
        pass
    else:
        # Z/nZ for n not prime
        assert raises(lambda: P([1, 2, 1]).gcd(P([1, 1])), DomainError)
        assert raises(lambda: P([1, 2, 1]).gcd(None), TypeError) # type: ignore

    if is_field:
        p1 = P([1, 0, 1])
        p2 = P([2, 1])
        g, s, t = P([1]), P([1])/5, P([2, -1])/5
        assert p1.xgcd(p2) == (g, s, t)
        assert raises(lambda: p1.xgcd(None), TypeError) # type: ignore

    if not composite_characteristic:
        assert P([1, 2, 1]).factor() == (S(1), [(P([1, 1]), 2)])
    elif nmod_poly_will_crash:
        pass
    else:
        assert raises(lambda: P([1, 2, 1]).factor(), DomainError)

    if not composite_characteristic:
        assert P([1, 2, 1]).sqrt() == P([1, 1])
        assert raises(lambda: P([1, 2, 2]).sqrt(), DomainError)
    elif nmod_poly_will_crash:
        pass
    else:
        assert raises(lambda: P([1, 2, 1]).sqrt(), DomainError)

    if P == flint.fmpq_poly:
        assert raises(lambda: (P([1, 2, 1])/3).sqrt(), ValueError)
        assert (P([1, 2, 1])/4).sqrt() == P([1, 1])/2

    assert P([]).deflation() == (P([]), 1)
    assert P([1, 2]).deflation() == (P([1, 2]), 1)
    assert P([1, 0, 2]).deflation() == (P([1, 2]), 2)

    assert P([1, 2, 1]).derivative() == P([2, 2])

    if is_field:
        p = P([1, 2, 1])
        if isinstance(p, (flint.fmpq_poly, flint.nmod_poly, flint.fmpz_mod_poly)):
            assert p.integral() == P([0, 1, 1, S(1)/3])
        elif isinstance(p, flint.fq_default_poly):
            assert raises(lambda: p.integral(), NotImplementedError) # type: ignore
        else:
            assert False

    # resultant checks.
    x = P([0, 1])

    if composite_characteristic and type(x) in [flint.fmpz_mod_poly, flint.nmod_poly]:
        # Flint sometimes crashes in this case, even though the resultant
        # could be computed.
        divisor = characteristic.factor()[0][0]
        assert raises(lambda: x.resultant(x + divisor), ValueError)
    elif type(x) == flint.fq_default_poly:
        # Flint does not implement resultants over GF(q) for nonprime q, so
        # there's nothing for us to check.
        pass
    else:
        assert x.resultant(x) == 0
        assert x.resultant(x**2 + x - x) == 0
        assert x.resultant(x**10 - x**5 + 1) == S(1)
        assert (x - 1).resultant(x**5 + 1) == S(2)

        assert (x**5 + 1).discriminant() == S(3125)
        assert (x**5 + 1).resultant(5 * x**4) == S(3125)

        for k in range(-10, 10):
            assert x.resultant(x + S(k)) == S(k)


def test_poly_resultants():
    # Check that the resultant of two cyclotomic polynomials is right.
    # See Dresden's 2012 "Resultants of Cyclotomic Polynomials"
    for m in range(1, 50):
        for n in range(m + 1, 50):
            a = flint.fmpz_poly.cyclotomic(m)
            b = flint.fmpz_poly.cyclotomic(n)
            q, r = divmod(flint.fmpz(n), flint.fmpz(m))
            fs = q.factor()
            if r != 0 or len(fs) > 1:
                assert a.resultant(b) == 1
            else:
                prime = fs[0][0]
                tot = flint.fmpz(m).euler_phi()
                assert a.resultant(b) == prime**tot


def _all_mpolys(): # -> _all_mpolys_type:
    return (
        (flint.fmpz_mpoly, flint.fmpz_mpoly_ctx.get, flint.fmpz, False, flint.fmpz(0)),
        (flint.fmpq_mpoly, flint.fmpq_mpoly_ctx.get, flint.fmpq, True, flint.fmpz(0)),
        (
            flint.fmpz_mod_mpoly,
            lambda *args, **kwargs: flint.fmpz_mod_mpoly_ctx.get(*args, **kwargs, modulus=101),
            lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(101)),
            True,
            flint.fmpz(101),
        ),
        (
            flint.fmpz_mod_mpoly,
            lambda *args, **kwargs: flint.fmpz_mod_mpoly_ctx.get(*args, **kwargs, modulus=100),
            lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(100)),
            False,
            flint.fmpz(100),
        ),
        (
            flint.nmod_mpoly,
            lambda *args, **kwargs: flint.nmod_mpoly_ctx.get(*args, **kwargs, modulus=101),
            lambda x: flint.nmod(x, 101),
            True,
            flint.fmpz(101),
        ),
        (
            flint.nmod_mpoly,
            lambda *args, **kwargs: flint.nmod_mpoly_ctx.get(*args, **kwargs, modulus=100),
            lambda x: flint.nmod(x, 100),
            False,
            flint.fmpz(100),
        ),
    )


class _GetMPolyCtx(Protocol[Tmpolyctx_p_co]):
    def __call__(self,
        names: Iterable[str | tuple[str, int]] | tuple[str, int],
        ordering: str | flint.Ordering = "lex"
    ) -> Tmpolyctx_p_co:
        ...


_MPolyTestCase = tuple[
    type[Tmpoly_p],
    _GetMPolyCtx[typ.mpoly_context_p[Tmpoly_p, Tscalar]],
    Callable[[int], Tscalar],
    bool,
    flint.fmpz
]


class _Q(Protocol[Tscalar_co]):
    def __call__(self, a: int, b: int | None = None, /) -> Tscalar_co:
        ...


def _is_Q(typ: object) -> TypeIs[_Q]:
    return typ is flint.fmpq


def _for_all_mpolys(test: Callable[[_MPolyTestCase], None]) -> None:
    """Test all mpoly types with the given test function."""
    # Spell it out like this so that a type checker can understand the types
    # in the generics for each call of test().
    test((
        flint.fmpz_mpoly,
        flint.fmpz_mpoly_ctx.get,
        flint.fmpz,
        False,
        flint.fmpz(0)
    ))
    test((
        flint.fmpq_mpoly,
        flint.fmpq_mpoly_ctx.get,
        flint.fmpq,
        True,
        flint.fmpz(0)
    ))
    test((
        flint.fmpz_mod_mpoly,
        lambda *args, **kwargs: flint.fmpz_mod_mpoly_ctx.get(*args, **kwargs, modulus=101),
        lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(101)),
        True,
        flint.fmpz(101),
    ))
    test((
        flint.fmpz_mod_mpoly,
        lambda *args, **kwargs: flint.fmpz_mod_mpoly_ctx.get(*args, **kwargs, modulus=100),
        lambda x: flint.fmpz_mod(x, flint.fmpz_mod_ctx(100)),
        False,
        flint.fmpz(100),
    ))
    test((
        flint.nmod_mpoly,
        lambda *args, **kwargs: flint.nmod_mpoly_ctx.get(*args, **kwargs, modulus=101),
        lambda x: flint.nmod(x, 101),
        True,
        flint.fmpz(101),
    ))
    test((
        flint.nmod_mpoly,
        lambda *args, **kwargs: flint.nmod_mpoly_ctx.get(*args, **kwargs, modulus=100),
        lambda x: flint.nmod(x, 100),
        False,
        flint.fmpz(100),
    ))


def all_mpolys(f: Callable[[_MPolyTestCase], None]) -> Callable[[], None]:
    """Decorator to run a test function for all mpoly types."""
    def wrapper():
        _for_all_mpolys(f)
    wrapper.__name__ = f.__name__
    return wrapper


@all_mpolys
def test_mpolys_constructor(args: _MPolyTestCase[Tmpoly_p, Tscalar]) -> None:
    P, get_context, S, _, _ = args

    ctx = get_context(("x", 2))

    def mpoly(x):
        return ctx.from_dict(x)

    def quick_poly():
        return mpoly({(0, 0): 1, (0, 1): 2, (1, 0): 3, (2, 2): 4})

    assert raises(lambda : ctx.__class__("x", flint.Ordering.lex), RuntimeError) # type: ignore
    assert raises(lambda: get_context(("x", 2), ordering="bad"), ValueError) # type: ignore
    assert raises(lambda: get_context(("x", -1)), ValueError)
    assert raises(lambda: ctx.constant("bad"), TypeError) # type: ignore
    assert raises(lambda: ctx.from_dict("bad"), ValueError) # type: ignore
    assert raises(lambda: ctx.from_dict({(0, 0): "bad"}), TypeError) # type: ignore
    assert raises(lambda: ctx.from_dict({(0, "bad"): 1}), TypeError) # type: ignore
    assert raises(lambda: ctx.from_dict({(0,): 1}), ValueError)
    assert raises(lambda: ctx.gen(-1), IndexError)
    assert raises(lambda: ctx.gen(10), IndexError)

    assert raises(lambda: P(val=get_context(("x",)).constant(0), ctx=ctx), IncompatibleContextError) # type: ignore
    assert raises(lambda: P(val={}, ctx=None), ValueError)
    assert raises(lambda: P(val={"bad": 1}, ctx=None), ValueError) # type: ignore
    assert raises(lambda: P(val="1", ctx=None), ValueError)

    ctx1 = get_context(("x", 4))
    ctx2 = get_context(("x", 4), ordering="deglex")
    assert ctx1.drop_gens(ctx1.names()).names() == tuple()
    assert ctx1.drop_gens((ctx1.name(1), ctx1.name(2))).names() == (ctx1.name(0), ctx1.name(3))
    assert ctx1.drop_gens(tuple()).names() == ctx1.names()
    assert ctx1.drop_gens((-1,)).names() == ctx1.names()[:-1]

    assert ctx.infer_generator_mapping(ctx) == {i: i for i in range(ctx.nvars())}
    assert ctx1.infer_generator_mapping(ctx) == {0: 0, 1: 1}
    assert ctx1.drop_gens(ctx.names()).infer_generator_mapping(ctx) == {}

    assert quick_poly().project_to_context(ctx1) == \
        ctx1.from_dict(
            {(0, 0, 0, 0): 1, (0, 1, 0, 0): 2, (1, 0, 0, 0): 3, (2, 2, 0, 0): 4}
        )
    new_poly = quick_poly().project_to_context(ctx1)
    assert ctx1.drop_gens(new_poly.unused_gens()) == ctx
    assert new_poly.project_to_context(ctx) == quick_poly()

    new_poly = quick_poly().project_to_context(ctx2)
    new_ctx = ctx2.drop_gens(new_poly.unused_gens())
    assert new_ctx != ctx
    assert new_poly != quick_poly()

    new_ctx = new_ctx.from_context(new_ctx, ordering=ctx.ordering())
    assert new_ctx == ctx
    assert new_poly.project_to_context(new_ctx) == quick_poly()

    assert ctx.append_gens(*ctx1.names()[-2:]) == ctx1

    assert P(val={(0, 0): 1}, ctx=ctx) == ctx.from_dict({(0, 0): 1})
    assert P(ctx=ctx).context() == ctx
    assert P(1, ctx=ctx).is_one() / 2
    assert ctx.gen(1) == ctx.from_dict({(0, 1): 1})

    assert ctx.nvars() == 2
    assert ctx.ordering() == flint.Ordering.lex

    ctx1 = get_context(("x", 4))
    assert [ctx1.name(i) for i in range(4)] == ['x0', 'x1', 'x2', 'x3']
    for order in list(flint.Ordering):
        ctx1 = get_context(("x", 4), ordering=order)
        assert ctx1.ordering() == order

    assert ctx.constant(1) == mpoly({(0, 0): 1}) == P(1, ctx=ctx)

    assert raises(lambda: P([None]), TypeError) # type: ignore
    assert raises(lambda: P(object()), TypeError) # type: ignore
    assert raises(lambda: P(None), TypeError) # type: ignore
    assert raises(lambda: P(None, None), TypeError) # type: ignore
    assert raises(lambda: P([1,2], None), TypeError) # type: ignore
    assert raises(lambda: P(1, None), ValueError) # type: ignore

    assert len(P(ctx=ctx)) == len(mpoly({(0, 0): 0})) == 0
    assert len(P(1, ctx=ctx)) == len(mpoly({(0, 0): 1})) == 1
    assert len(mpoly({(0, 0): 1, (0, 1): 1})) == 2
    assert len(mpoly({(0, 0): 1, (0, 1): 1, (1, 0): 1})) == 3

    # degree is -1 when empty poly
    assert P(ctx=ctx).degrees() == mpoly({(0, 0): 0}).degrees() == (-1, -1)
    assert P(1, ctx=ctx).degrees() == mpoly({(0, 0): 1}).degrees() == (0, 0)
    assert mpoly({(0, 0): 1, (0, 1): 1}).degrees() == (0, 1)
    assert mpoly({(0, 0): 1, (0, 1): 1, (1, 0): 1}).degrees() == (1, 1)
    assert mpoly({(0, 0): 1, (0, 1): 1, (1, 0): 1, (2, 2): 2}).degrees() == (2, 2)

    assert (P(1, ctx=ctx) == P(1, ctx=ctx)) is True
    assert (P(1, ctx=ctx) != P(1, ctx=ctx)) is False
    assert (P(1, ctx=ctx) == P(2, ctx=ctx)) is False
    assert (P(1, ctx=ctx) != P(2, ctx=ctx)) is True

    assert (P(1, ctx=ctx) == 1) is True
    assert (P(1, ctx=ctx) != 1) is False
    assert (1 == P(1, ctx=ctx)) is True
    assert (1 != P(1, ctx=ctx)) is False

    assert (P(1, ctx=ctx) == S(1)) is True
    assert (P(1, ctx=ctx) != S(1)) is False
    assert (S(1) == P(1, ctx=ctx)) is True
    assert (S(1) != P(1, ctx=ctx)) is False

    assert (P(1, ctx=ctx) == P(1, ctx=ctx1)) is False
    assert (P(1, ctx=ctx) != P(1, ctx=ctx1)) is True

    assert (P(1, ctx=ctx) == None) is False
    assert (P(1, ctx=ctx) != None) is True
    assert (None == P(1, ctx=ctx)) is False
    assert (None != P(1, ctx=ctx)) is True

    assert P(ctx.from_dict({(0, 1): 3})) == ctx.from_dict({(0, 1): 3})
    assert P({(0, 1): 3}, ctx=ctx) == ctx.from_dict({(0, 1): 3})

    if P is flint.fmpq_mpoly and isinstance(ctx, flint.fmpq_mpoly_ctx):
        ctx_z = flint.fmpz_mpoly_ctx.get((("x", 2),))
        assert quick_poly() == P(ctx_z.from_dict({(0, 0): 1, (0, 1): 2, (1, 0): 3, (2, 2): 4}))
        assert P(ctx_z.from_dict({(0, 0): 1}), ctx=ctx) == P({(0, 0): 1}, ctx=ctx)


@all_mpolys
def test_mpolys_properties(args: _MPolyTestCase[Tmpoly_p, Tscalar]) -> None:

    P, get_context, S, is_field, characteristic = args

    # Division under modulo will raise a flint exception if something is
    # not invertible, crashing the program. We can't tell before what is
    # invertible and what is not before hand so we always raise an
    # exception, except for fmpz_mpoly, that returns an bool noting if the
    # division is exact or not.
    composite_characteristic = characteristic != 0 and not characteristic.is_prime()

    def mpoly(x):
        return ctx.from_dict(x)

    def quick_poly():
        return mpoly({(0, 0): 1, (0, 1): 2, (1, 0): 3, (2, 2): 4})

    ctx = get_context(("x", 2))
    ctx1 = get_context(("x", 4))

    assert raises(lambda: P(ctx=ctx) < P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) <= P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) > P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) >= P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) < None, TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) <= None, TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) > None, TypeError) # type: ignore
    assert raises(lambda: P(ctx=ctx) >= None, TypeError) # type: ignore
    assert raises(lambda: None < P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: None <= P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: None > P(ctx=ctx), TypeError) # type: ignore
    assert raises(lambda: None >= P(ctx=ctx), TypeError) # type: ignore

    p = quick_poly()
    assert p.coefficient(2) == S(2)
    assert raises(lambda: p.coefficient(-1), IndexError)
    assert raises(lambda: p.coefficient(10), IndexError)

    assert raises(lambda: p[-1], TypeError) # type: ignore
    assert raises(lambda: p[4], TypeError) # type: ignore

    assert p[(2, 2)] == 4
    assert p[(0, 0)] == 1
    assert raises(lambda: p[(1,)], ValueError)
    assert raises(lambda: p[(1, "bad")], TypeError) # type: ignore
    assert raises(lambda: p["bad"], TypeError) # type: ignore

    p = quick_poly()
    p[(1, 0)] = S(10)
    assert p == mpoly({(0, 0): 1, (0, 1): 2, (1, 0): 10, (2, 2): 4})

    p = quick_poly()
    p[(1, 0)] = p[(1, 0)]
    assert p == quick_poly()
    assert (1, 0) in p
    assert (100, 100) not in p

    assert raises(lambda: p.__setitem__((4,), 1), ValueError)

    assert raises(lambda: p.__setitem__((1,), 1), ValueError)
    assert raises(lambda: p.__setitem__((1, "bad"), 1), TypeError) # type: ignore
    assert raises(lambda: p.__setitem__(("bad", 1), 1), TypeError) # type: ignore

    assert raises(lambda: p.__setitem__((2, 1), None), TypeError) # type: ignore

    assert P(ctx=ctx).repr() == f"{ctx.__class__.__name__}(2, '<Ordering.lex: 'lex'>', ('x0', 'x1')).from_dict({{}})"
    assert P(1, ctx=ctx).repr() == f"{ctx.__class__.__name__}(2, '<Ordering.lex: 'lex'>', ('x0', 'x1')).from_dict({{(0, 0): 1}})"
    assert str(quick_poly()) == repr(quick_poly()) == '4*x0^2*x1^2 + 3*x0 + 2*x1 + 1'

    assert p.monomial(0) == (2, 2)
    assert p.monomial(3) == (0, 0)
    assert raises(lambda: p.monomial(-1), IndexError)
    assert raises(lambda: p.monomial(4), IndexError)

    assert p.total_degree() == 4
    assert P(ctx=ctx).total_degree() == -1
    assert P(1, ctx=ctx).total_degree() == 0

    p = quick_poly()
    assert p(0, 0) == p(0, S(0)) == p(S(0), S(0)) == S(1) == 1
    assert p(1, 1) == S(10) == 10

    p = quick_poly()
    assert p.monoms() == [(2, 2), (1, 0), (0, 1), (0, 0)]
    assert p.coeffs() == [4, 3, 2, 1]
    assert list(p.terms()) == list(zip([(2, 2), (1, 0), (0, 1), (0, 0)], [4, 3, 2, 1]))

    assert p.subs({"x1": S(0), 0: S(0)}) == ctx.from_dict({(0, 0): 1})
    assert p.compose(p.subs({"x1": 0}), ctx.from_dict({(0, 1): 1})) == mpoly({
        (2, 2): 36,
        (1, 2): 24,
        (1, 0): 9,
        (0, 2): 4,
        (0, 1): 2,
        (0, 0): 4
    })
    assert p.compose(ctx.from_dict({(1, 0): 1}), ctx.from_dict({(0, 1): 1})) == p

    assert raises(lambda: p(None, None), TypeError) # type: ignore
    assert raises(lambda: p(1), ValueError) # type: ignore
    assert raises(lambda: p(0, 1, 2), ValueError) # type: ignore

    assert raises(lambda: p.subs({"x0": None}), TypeError) # type: ignore
    assert raises(lambda: p.subs({"x0": None, "x1": None}), TypeError) # type: ignore
    assert raises(lambda: p.subs({"a": 1}), ValueError)
    assert raises(lambda: p.subs({"x0": 0, "x1": 1, "x2": 2}), ValueError)

    no_gens_ctx = get_context(tuple())
    no_gens_p = P("2", no_gens_ctx)
    assert no_gens_p.compose(ctx=ctx1).context() is ctx1
    assert raises(lambda: no_gens_p.compose(), ValueError)

    assert raises(lambda: p.compose(p, P(ctx=ctx1)), IncompatibleContextError)

    assert bool(P(ctx=ctx)) is False
    assert bool(P(1, ctx=ctx)) is True

    assert P(ctx=ctx).is_zero() is True
    assert P(1, ctx=ctx).is_zero() is False

    assert P(ctx=ctx).is_one() is False
    assert P(1, ctx=ctx).is_one() is True

    assert +quick_poly() \
        == quick_poly()

    assert -quick_poly() == mpoly({(0, 0): -1, (0, 1): -2, (1, 0): -3, (2, 2): -4})

    assert quick_poly() \
        + mpoly({(0, 0): 5, (0, 1): 6, (1, 0): 7, (2, 2): 8}) \
        == mpoly({(0, 0): 6, (0, 1): 8, (1, 0): 10, (2, 2): 12})

    for T in [int, S, flint.fmpz, lambda x: P(x, ctx=ctx)]:
        p = quick_poly()
        p += T(1)
        q = quick_poly()
        assert q.iadd(T(1)) is None
        assert quick_poly() + T(1) \
            == p == q == mpoly({(0, 0): 2, (0, 1): 2, (1, 0): 3, (2, 2): 4})
        assert T(1) + quick_poly() \
            == mpoly({(0, 0): 2, (0, 1): 2, (1, 0): 3, (2, 2): 4})

    assert raises(lambda: mpoly({(0, 0): 2, (0, 1): 2, (1, 0): 3, (2, 2): 4}) + None, TypeError) # type: ignore
    assert raises(lambda: None + mpoly({(0, 0): 2, (0, 1): 2, (1, 0): 3, (2, 2): 4}), TypeError) # type: ignore
    assert raises(lambda: quick_poly() + P(ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: quick_poly().iadd(P(ctx=ctx1)), IncompatibleContextError)
    assert raises(lambda: quick_poly().iadd(None), NotImplementedError) # type: ignore

    assert quick_poly() - mpoly({(0, 0): 5, (0, 1): 6, (1, 0): 7, (2, 2): 8}) \
        == mpoly({(0, 0): -4, (0, 1): -4, (1, 0): -4, (2, 2): -4})

    for T in [int, S, flint.fmpz, lambda x: P(x, ctx=ctx)]:
        p = quick_poly()
        p -= T(1)
        q = quick_poly()
        assert q.isub(T(1)) is None
        assert quick_poly() - T(1) == p == q == mpoly({(0, 1): 2, (1, 0): 3, (2, 2): 4})
        assert T(1) - quick_poly() == mpoly({(0, 1): -2, (1, 0): -3, (2, 2): -4})

    assert raises(lambda: quick_poly() - None, TypeError) # type: ignore
    assert raises(lambda: None - quick_poly(), TypeError) # type: ignore
    assert raises(lambda: quick_poly() - P(ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: quick_poly().isub(P(ctx=ctx1)), IncompatibleContextError)
    assert raises(lambda: quick_poly().isub(None), NotImplementedError) # type: ignore

    assert quick_poly() * mpoly({(1, 0): 5, (0, 1): 6}) \
        == mpoly({
            (3, 2): 20,
            (2, 3): 24,
            (2, 0): 15,
            (1, 1): 28,
            (1, 0): 5,
            (0, 2): 12,
            (0, 1): 6
        })

    for T in [int, S, flint.fmpz, lambda x: P(x, ctx=ctx)]:
        p = quick_poly()
        p *= T(2)
        q = quick_poly()
        assert q.imul(T(2)) is None
        assert quick_poly() * T(2) == p == q == mpoly({(0, 0): 2, (0, 1): 4, (1, 0): 6, (2, 2): 8})
        assert T(2) * quick_poly() == mpoly({(0, 0): 2, (0, 1): 4, (1, 0): 6, (2, 2): 8})

    assert raises(lambda: quick_poly() * None, TypeError) # type: ignore
    assert raises(lambda: None * quick_poly(), TypeError) # type: ignore
    assert raises(lambda: quick_poly() * P(ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: quick_poly().imul(P(ctx=ctx1)), IncompatibleContextError)
    assert raises(lambda: quick_poly().imul(None), NotImplementedError) # type: ignore

    if composite_characteristic:
        assert raises(lambda: quick_poly() // mpoly({(1, 1): 1}), DomainError)
        assert raises(lambda: quick_poly() % mpoly({(1, 1): 1}), DomainError)
        assert raises(lambda: divmod(quick_poly(), mpoly({(1, 1): 1})), DomainError)
    else:
        assert quick_poly() // mpoly({(1, 1): 1}) == mpoly({(1, 1): 4})
        assert quick_poly() % mpoly({(1, 1): 1}) \
            == mpoly({(1, 0): 3, (0, 1): 2, (0, 0): 1})
        assert divmod(quick_poly(), mpoly({(1, 1): 1})) \
            == (mpoly({(1, 1): 4}), mpoly({(1, 0): 3, (0, 1): 2, (0, 0): 1}))

        assert 1 / P(1, ctx=ctx) == P(1, ctx=ctx)
        assert quick_poly() / 1 == quick_poly()
        assert quick_poly() // 1 == quick_poly()
        assert quick_poly() % 1 == P(ctx=ctx)
        assert divmod(quick_poly(), 1) == (quick_poly(), P(ctx=ctx))

        assert S(1) / P(1, ctx=ctx) == P(1, ctx=ctx)
        assert quick_poly() / S(1) == quick_poly()
        assert quick_poly() // S(1) == quick_poly()
        assert quick_poly() % S(1) == P(ctx=ctx)
        assert divmod(quick_poly(), S(1)) == (quick_poly(), P(ctx=ctx))

    if is_field:
        assert quick_poly() / 3 == mpoly({(0, 0): S(1) / 3, (0, 1): S(2) / 3, (1, 0): S(1), (2, 2): S(4) / 3})
    else:
        assert raises(lambda: quick_poly() / 3, DomainError)

    f = mpoly({(1, 1): 4, (0, 0): 1})
    g = mpoly({(0, 1): 2, (1, 0): 2})
    if not composite_characteristic:
        assert 1 // quick_poly() == P(ctx=ctx)
        assert 1 % quick_poly() == P(1, ctx=ctx)
        assert divmod(1, quick_poly()) == (P(ctx=ctx), P(1, ctx=ctx))

        assert S(1) // quick_poly() == P(ctx=ctx)
        assert S(1) % quick_poly() == P(1, ctx=ctx)
        assert divmod(S(1), quick_poly()) == (P(ctx=ctx), P(1, ctx=ctx))

        assert f * g / mpoly({(0, 1): 1, (1, 0): 1}) \
            == mpoly({(1, 1): 8, (0, 0): 2})

        if not is_field:
            assert raises(lambda: 1 / quick_poly(), DomainError)
            assert raises(lambda: quick_poly() / P(2, ctx=ctx), DomainError)

    # We prefer various other errors to the "division not supported" domain error so these are safe.
    assert raises(lambda: quick_poly() / None, TypeError) # type: ignore
    assert raises(lambda: quick_poly() // None, TypeError) # type: ignore
    assert raises(lambda: quick_poly() % None, TypeError) # type: ignore
    assert raises(lambda: divmod(quick_poly(), None), TypeError) # type: ignore

    assert raises(lambda: None / quick_poly(), TypeError) # type: ignore
    assert raises(lambda: None // quick_poly(), TypeError) # type: ignore
    assert raises(lambda: None % quick_poly(), TypeError) # type: ignore
    assert raises(lambda: divmod(None, quick_poly()), TypeError) # type: ignore

    assert raises(lambda: quick_poly() / 0, ZeroDivisionError)
    assert raises(lambda: quick_poly() // 0, ZeroDivisionError)
    assert raises(lambda: quick_poly() % 0, ZeroDivisionError)
    assert raises(lambda: divmod(quick_poly(), 0), ZeroDivisionError)

    assert raises(lambda: 1 / P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: 1 // P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: 1 % P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: divmod(1, P(ctx=ctx)), ZeroDivisionError)

    assert raises(lambda: quick_poly() / P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: quick_poly() // P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: quick_poly() % P(ctx=ctx), ZeroDivisionError)
    assert raises(lambda: divmod(quick_poly(), P(ctx=ctx)), ZeroDivisionError)

    assert raises(lambda: quick_poly() / P(1, ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: quick_poly() // P(1, ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: quick_poly() % P(1, ctx=ctx1), IncompatibleContextError)
    assert raises(lambda: divmod(quick_poly(), P(1, ctx=ctx1)), IncompatibleContextError)

    assert quick_poly() ** 0 == P(1, ctx=ctx)
    assert quick_poly() ** 1 == quick_poly()
    assert quick_poly() ** 2 == mpoly({
        (4, 4): 16,
        (3, 2): 24,
        (2, 3): 16,
        (2, 2): 8,
        (2, 0): 9,
        (1, 1): 12,
        (1, 0): 6,
        (0, 2): 4,
        (0, 1): 4,
        (0, 0): 1,
    })
    assert raises(lambda: P(ctx=ctx) ** -1, ZeroDivisionError)
    assert raises(lambda: P(ctx=ctx) ** None, TypeError) # type: ignore

    # # XXX: Not sure what this should do in general:
    assert raises(lambda: pow(P(1, ctx=ctx), 2, 3), NotImplementedError) # type: ignore

    if composite_characteristic:
        assert raises(lambda: (f * g).gcd(f), DomainError)
    else:
        if is_field:
            assert (f * g).gcd(f) == f / 4
        else:
            assert (f * g).gcd(f) == f
        assert raises(lambda: quick_poly().gcd(None), TypeError) # type: ignore
        assert raises(lambda: quick_poly().gcd(P(ctx=ctx1)), IncompatibleContextError)

    x0, x1 = ctx.gens()
    assert (2*x0**3 + 4*x0**2 + 6*x0).term_content() == x0 if is_field else 2*x0
    assert (3*x0**2*x1 + 6*x0*x1**2 + 9*x1**3).term_content() == x1 if is_field else 3*x1

    assert (x0**2 - 1).resultant(x0**2 - 2*x0 + 1, 'x0') == 0
    assert (x0**2 + x1**2 - 1).resultant(x0 - x1, 'x0') == 2*x1**2 - 1

    assert (x0**3 - 6*x0**2 + 11*x0 - 6).discriminant('x0') == 4
    assert (x0**2 + 4*x0*x1 + 4*x1**2 - 1).discriminant('x0') == 4

    f1 = 3*x0**2*x1**2 + 6*x0*x1**2 + 9*x1**2
    res, stride = f1.deflation()
    assert res == 3*x0**2*x1 + 6*x0*x1 + 9*x1
    assert tuple(stride) == (1, 2)

    g1 = ((x0**2 + x1**2)**3 + (x0**2 + x1**2)**2 + 1)
    res, stride = g1.deflation()
    assert res == x0**3 + 3*x0**2*x1 + x0**2 + 3*x0*x1**2 + 2*x0*x1 + x1**3 + x1**2 + 1
    assert tuple(stride) == (2, 2)

    for p in [f1, g1]:
        pd, n = p.deflation()
        assert pd.inflate(n) == p
        assert p.deflate(n).inflate(n) == p

        pd, n, m = p.deflation_monom()
        assert m * pd.inflate(n) == p

        if not composite_characteristic:
            n, i = p.deflation_index()
            m = ctx.term(exp_vec=i)
            assert (p / m).deflate(n).inflate(n) * m == p

    if isinstance(x0, flint.fmpz_mpoly) and isinstance(x1, flint.fmpz_mpoly):
        assert (x0**2 * x1 + x0 * x1).primitive() == (1, x0**2*x1 + x0*x1)
        assert (4 * x0 + 2 * x0 * x1).primitive() == (2, x0 * x1 + 2 * x0)

    if composite_characteristic:
        # Factorisation not allowed over Z/nZ for n not prime.
        # Flint would abort so we raise an exception instead:
        assert raises(lambda: (f * g).factor(), DomainError)
    elif characteristic == 0:
        # Primitive factors over Z for fmpz_mpoly and fmpq_mpoly
        assert (f * g).factor() == (S(2), [(g / 2, 1), (f, 1)])
    elif is_field:
        # Monic polynomials over Z/pZ for nmod_mpoly and fmpz_mod_mpoly
        assert (f * g).factor() == (S(8), [(g / 2, 1), (f / 4, 1)])

    if composite_characteristic:
        assert raises(lambda: (f * g).sqrt(), DomainError)
    else:
        assert (f * f).sqrt() == f
        if isinstance(f, flint.fmpz_mpoly):
            assert (f * f).sqrt(assume_perfect_square=True) == f
        assert raises(lambda: quick_poly().sqrt(), DomainError)

    p = quick_poly()
    assert p.derivative(0) == p.derivative("x0") == mpoly({(0, 0): 3, (1, 2): 8})
    assert p.derivative(1) == p.derivative("x1") == mpoly({(0, 0): 2, (2, 1): 8})

    assert raises(lambda: p.derivative("x3"), ValueError)
    assert raises(lambda: p.derivative(3), IndexError)
    assert raises(lambda: p.derivative(None), TypeError) # type: ignore

    if isinstance(p, (flint.fmpz_mpoly, flint.fmpq_mpoly)):
        if isinstance(p, flint.fmpq_mpoly) and _is_Q(S):
            assert p.integral(0) == p.integral("x0") == \
                mpoly({(3, 2): S(4, 3), (2, 0): S(3, 2), (1, 1): S(2), (1, 0): S(1)})
            assert p.integral(1) == p.integral("x1") == \
                mpoly({(2, 3): S(4, 3), (1, 1): S(3), (0, 2): S(1), (0, 1): S(1)})
        else:
            assert p.integral(0) == p.integral("x0") == \
                (6, mpoly({(3, 2): 8, (2, 0): 9, (1, 1): 12, (1, 0): 6}))
            assert p.integral(1) == p.integral("x1") == \
                (3, mpoly({(2, 3): 4, (1, 1): 9, (0, 2): 3, (0, 1): 3}))

        assert raises(lambda: p.integral("x3"), ValueError)
        assert raises(lambda: p.integral(3), IndexError)
        assert raises(lambda: p.integral(None), TypeError) # type: ignore

    elif isinstance(p, (flint.nmod_mpoly, flint.fmpz_mod_mpoly)):
        pass # No integral method
    else:
        assert False # New _mpoly types?


def _all_mpoly_vecs():
    return [(flint.fmpz_mpoly_ctx, flint.fmpz_mpoly_vec), (flint.fmpq_mpoly_ctx, flint.fmpq_mpoly_vec)]


def test_fmpz_mpoly_vec():
    for context, mpoly_vec in _all_mpoly_vecs():
        has_groebner_functions = mpoly_vec is flint.fmpz_mpoly_vec

        ctx = context.get((("x", 2),))
        ctx1 = context.get((("x", 4),))
        x, y = ctx.gens()

        vec = mpoly_vec(3, ctx)

        assert vec[0] == ctx.from_dict({})
        assert vec[1] == ctx.from_dict({})

        assert vec[2] == ctx.from_dict({})

        assert len(vec) == 3
        assert str(vec) == f"[{', '.join(str(ctx.from_dict({})) for _ in range(3))}]"
        assert repr(vec) == f"{mpoly_vec.__name__}([{', '.join(str(ctx.from_dict({})) for _ in range(3))}], ctx={str(ctx)})"

        assert raises(lambda: vec[None], TypeError) # type: ignore
        assert raises(lambda: vec[-1], IndexError)

        vec[1] = x * y
        assert vec == mpoly_vec([ctx.from_dict({}), x * y, ctx.from_dict({})], ctx)
        assert vec != mpoly_vec([x, x * y, ctx.from_dict({})], ctx)
        assert vec != mpoly_vec([ctx.from_dict({})], ctx)
        assert vec != mpoly_vec([ctx1.from_dict({})], ctx1)
        assert tuple(vec) == tuple(mpoly_vec([ctx.from_dict({}), x * y, ctx.from_dict({})], ctx))
        assert raises(lambda: vec.__setitem__(None, 0), TypeError) # type: ignore
        assert raises(lambda: vec.__setitem__(-1, 0), IndexError) # type: ignore
        assert raises(lambda: vec.__setitem__(0, 0), TypeError) # type: ignore
        assert raises(lambda: vec.__setitem__(0, ctx1.from_dict({})), IncompatibleContextError)

        if has_groebner_functions:
            ctx = context.get(("x", "y", "z"))

            # Examples here cannibalised from
            # https://en.wikipedia.org/wiki/Gr%C3%B6bner_basis#Example_and_counterexample
            x, y, z = ctx.gens()
            f = x**2 - y
            f2 = 3 * x**2 - y
            g = x**3 - x
            g2 = x**3 * y - x
            k = x * y - x
            k2 = 3 * x * y - 3 * x
            h = y**2 - y

            vec = mpoly_vec([f, k, h], ctx)
            assert vec.is_groebner()
            assert vec.is_groebner(mpoly_vec([f, g], ctx))
            assert not vec.is_groebner(mpoly_vec([f, x**3], ctx))

            assert mpoly_vec([f2, k, h], ctx).is_autoreduced()
            assert not mpoly_vec([f2, k2, h], ctx).is_autoreduced()

            vec = mpoly_vec([f2, k2, h], ctx)
            vec2 = vec.autoreduction()
            assert not vec.is_autoreduced()
            assert vec2.is_autoreduced()
            assert vec2 == mpoly_vec([3 * x**2 - y, x * y - x, y**2 - y], ctx)

            vec = mpoly_vec([f, g2], ctx)
            assert not vec.is_groebner()
            assert vec.buchberger_naive() == mpoly_vec([x**2 - y, x**3 * y - x, x * y**2 - x, y**3 - y], ctx)
            assert vec.buchberger_naive(limits=(2, 2, 512)) == (mpoly_vec([x**2 - y, x**3 * y - x], ctx), False)

            unreduced_basis = mpoly_vec([x**2 - 2, y**2 - 3, z - x - y], ctx).buchberger_naive()
            assert list(unreduced_basis) == [
                x**2 - 2,
                y**2 - 3,
                x + y - z,
                2*y*z - z**2 - 1,
                2*y + z**3 - 11*z,
                z**4 - 10*z**2 + 1
            ]

            assert list(unreduced_basis.autoreduction()) == [
                z**4 - 10 * z**2 + 1,
                2*y + z**3 - 11 * z,
                2 * x - z**3 + 9 * z
            ]


def _all_polys_mpolys():

    for P, S, is_field, characteristic in _all_polys():
        x = P([0, 1])
        y = None
        assert isinstance(x, (
            flint.fmpz_poly,
            flint.fmpq_poly,
            flint.nmod_poly,
            flint.fmpz_mod_poly,
            flint.fq_default_poly,
        ))
        yield P, S, [x, y], is_field, characteristic

    for P, get, S, is_field, characteristic in _all_mpolys():
        ctx = get(("x", "y"))
        x, y = ctx.gens()
        assert isinstance(x, (
            flint.fmpz_mpoly,
            flint.fmpq_mpoly,
            flint.nmod_mpoly,
            flint.fmpz_mod_mpoly,
        ))
        yield P, S, [x, y], is_field, characteristic


def test_properties_poly_mpoly():
    """Test is_zero, is_one etc for all polynomials."""
    for _, _, [x, _], _, _ in _all_polys_mpolys():

        zero = 0*x
        one = zero + 1
        two = one + 1

        assert zero.is_zero() is True
        assert one.is_zero() is False
        assert two.is_zero() is False
        assert x.is_zero() is False

        assert zero.is_one() is False
        assert one.is_one() is True
        assert two.is_one() is False
        assert x.is_one() is False

        assert zero.is_constant() is True
        assert one.is_constant() is True
        assert two.is_constant() is True
        assert x.is_constant() is False

        # is_gen?


def test_factor_poly_mpoly():
    """Test that factor() is consistent across different poly/mpoly types."""

    def check(p, coeff, factors):
        # Check all the types
        lc = p.leading_coefficient()
        assert type(coeff) is type(lc)
        assert isinstance(factors, list)
        assert all(isinstance(f, tuple) for f in factors)
        for fac, m in factors:
            assert type(fac) is type(p)
            assert type(m) is int

        # Check the actual factorisation!
        res = coeff
        for fac, m in factors:
            res *= fac ** m
        assert res == p

    def sort(factors):
        def sort_key(p):
            fac, m = p
            return (m, sorted(str(i) for i in fac.coeffs()))
        return sorted(factors, key=sort_key)

    def factor(p):
        coeff, factors = p.factor()
        check(p, coeff, factors)
        return coeff, sort(factors)

    def factor_sqf(p):
        coeff, factors = p.factor_squarefree()
        check(p, coeff, factors)
        return coeff, sort(factors)

    for P, S, [x, y], is_field, characteristic in _all_polys_mpolys():

        if characteristic != 0 and not characteristic.is_prime():
            # nmod_poly crashes for many operations with non-prime modulus
            #     https://github.com/flintlib/python-flint/issues/124
            # so we can't even test it...
            nmod_poly_will_crash = type(x) is flint.nmod_poly
            if nmod_poly_will_crash:
                continue

            try:
                assert S(4).sqrt() ** 2 == S(4)
            except DomainError:
                pass
            assert raises(lambda: (x**2).sqrt(), DomainError)
            assert raises(lambda: x.gcd(x), DomainError)
            assert raises(lambda: x.gcd(None), TypeError) # type: ignore
            assert raises(lambda: x.factor(), DomainError)
            assert raises(lambda: x.factor_squarefree(), DomainError)

            # All tests below can be expected to raise DomainError
            # Not sure if that is guaranteed in all cases though...
            continue

        assert S(0).sqrt() == S(0)
        assert S(1).sqrt() == S(1)
        assert S(4).sqrt()**2 == S(4)

        for i in range(-100, 100):
            try:
                assert S(i).sqrt() ** 2 == S(i)
            except DomainError:
                pass

        if characteristic == 0:
            assert raises(lambda: S(-1).sqrt(), DomainError)

        assert (0*x).sqrt() == 0*x
        assert (1*x/x).sqrt() == 0*x + 1
        assert (4*x/x).sqrt()**2 == 0*x + 4

        for i in range(-100, 100):
            try:
                assert (i*x).sqrt() ** 2 == i*x
            except DomainError:
                pass

        assert (x**2).sqrt() == x
        assert (S(4)*x**2).sqrt()**2 == S(4)*x**2
        assert raises(lambda: (x**2 + 1).sqrt(), DomainError)

        assert factor(0*x) == (S(0), [])
        assert factor(0*x + 1) == (S(1), [])
        assert factor(0*x + 3) == (S(3), [])
        assert factor(x) == (S(1), [(x, 1)])
        assert factor(-x) == (S(-1), [(x, 1)])
        assert factor(x**2) == (S(1), [(x, 2)])
        assert factor(2*(x+1)) == (S(2), [(x+1, 1)])

        assert factor_sqf(0*x) == (S(0), [])
        assert factor_sqf(0*x + 1) == (S(1), [])
        assert factor_sqf(0*x + 3) == (S(3), [])
        assert factor_sqf(-x) == (S(-1), [(x, 1)])
        assert factor_sqf(x**2) == (S(1), [(x, 2)])
        assert factor_sqf(2*(x+1)) == (S(2), [(x+1, 1)])

        assert (0*x).gcd(0*x) == 0*x
        assert (0*x).gcd(0*x + 1) == S(1)

        if not is_field:
            assert (0*x).gcd(0*x + 3) == S(3)
        else:
            assert (0*x).gcd(0*x + 3) == S(1)

        assert (2*x).gcd(x) == x
        assert (2*x).gcd(x**2) == x
        assert (2*x).gcd(x**2 + 1) == S(1)

        if not is_field:
            # primitive gcd over Z
            assert (2*x).gcd(4*x**2) == 2*x
        else:
            # monic gcd over Q, Z/pZ and GF(p^d)
            assert (2*x).gcd(4*x**2) == x

        if is_field and y is None:
            # xgcd is defined and consistent for all univariate polynomials
            # over a field (Q, Z/pZ, GF(p^d)).
            assert (2*x).xgcd(4*x) == (x, P(0), P(1)/4)
            assert (2*x).xgcd(4*x**2+1) == (P(1), -2*x, P(1))

        # mpoly types have a slightly different squarefree factorisation
        # because they handle trivial factors differently. It looks like a
        # monomial gcd is extracted but not recombined so the square-free
        # factors might not have unique multiplicities.
        #
        # Maybe it is worth making them consistent by absorbing the power
        # of x into a factor of equal multiplicity.
        assert factor(x*(x+1)) == (S(1), [(x, 1), (x+1, 1)])
        if y is None:
            # *_poly types
            assert factor_sqf(x*(x+1)) == (S(1), [(x**2+x, 1)])
        else:
            # *_mpoly types
            assert factor_sqf(x*(x+1)) == (S(1), [(x, 1), (x+1, 1)])

        # This is the same for all types because the extracted monomial has
        # a unique multiplicity.
        assert factor_sqf(x**2*(x+1)) == (S(1), [(x+1, 1), (x, 2)])

        # This is the same for all types because there is no trivial monomial
        # factor to extract.
        assert factor((x-1)*(x+1)) == (S(1), sort([(x-1, 1), (x+1, 1)]))
        assert factor_sqf((x-1)*(x+1)) == (S(1), [(x**2-1, 1)])

        # Some finite fields have sqrt(-1) so we can factor x**2 + 1
        try:
            i = S(-1).sqrt()
        except DomainError:
            i = None

        p = 3*(x-1)**2*(x+1)**2*(x**2 + 1)**3
        assert factor_sqf(p) == (S(3), [(x**2 - 1, 2), (x**2 + 1, 3)])

        if i is not None:
            assert factor(p) == (S(3), sort([(x+1, 2), (x-1, 2), (x+i, 3), (x-i, 3)]))
        else:
            assert factor(p) == (S(3), sort([(x-1, 2), (x+1, 2), (x**2+1, 3)]))

        if characteristic == 0:
            # primitive factors over Z for Z and Q.
            assert factor(2*x+1) == (S(1), [(2*x+1, 1)])
        else:
            # monic factors over Z/pZ and GF(p^d)
            assert factor(2*x+1) == (S(2), [(x+S(1)/2, 1)])

        if is_field:
            if characteristic == 0:
                assert factor((2*x+1)/7) == (S(1)/7, [(2*x+1, 1)])
            else:
                assert factor((2*x+1)/7) == (S(2)/7, [(x+S(1)/2, 1)])

        if y is not None:

            # *_mpoly types

            assert factor(x*y+1) == (S(1), [(x*y+1, 1)])
            assert factor(x*y) == (S(1), [(x, 1), (y, 1)])

            assert factor_sqf((x*y+1)**2*(x*y-1)) == (S(1), [(x*y-1, 1), (x*y+1, 2)])

            p = 2*x + y
            if characteristic == 0:
                assert factor(p) == factor_sqf(p) == (S(1), [(p, 1)])
            else:
                assert factor(p) == factor_sqf(p) == (S(2), [(p/2, 1)])

            if is_field:
                p = (2*x + y)/7
                if characteristic == 0:
                    assert factor(p) == factor_sqf(p) == (S(1)/7, [(7*p, 1)])
                else:
                    assert factor(p) == factor_sqf(p) == (S(2)/7, [(7*p/2, 1)])

            if not is_field:
                # primitive gcd over Z
                assert (2*(x+y)).gcd(4*(x+y)**2) == 2*(x+y)
            else:
                # monic gcd over Q, Z/pZ and GF(p^d)
                assert (2*(x+y)).gcd(4*(x+y)**2) == x + y


def test_division_poly_mpoly():
    """Test that division is consistent across different poly/mpoly types."""

    Z = flint.fmpz

    for _, S, [x, _], is_field, characteristic in _all_polys_mpolys():

        if characteristic != 0 and not characteristic.is_prime():
            # nmod_poly crashes for many operations with non-prime modulus
            #     https://github.com/flintlib/python-flint/issues/124
            # so we can't even test it...
            nmod_poly_will_crash = type(x) is flint.nmod_poly
            if nmod_poly_will_crash:
                continue

        one = x**0 # 1 as a polynomial
        two = one + one

        if is_field or characteristic == 0:
            assert x / x == x**0 == 1 == one
            assert x / 1 == x / S(1) == x / one == x**1 == x
            assert 1 / one == one**-1 == one**Z(-1) == 1, type(one)
            assert -1 / one == 1 / -one == (-one)**-1 == (-one)**Z(-1) == -one == -1
            assert (-one) ** -2 == (-one)**Z(-2) == one
            assert raises(lambda: 1 / x, DomainError)
            assert raises(lambda: x ** -1, DomainError)

        if is_field:
            half = S(1)/2 * one # 1/2 as a polynomial
            assert half == S(1)/2
            assert x / half == 2*x
            assert 1 / half == S(1) / half == one / half == one / (S(1)/2) == 2
            assert half ** -1 == half ** Z(-1) == 2
            assert two ** -1 == two ** Z(-1) == half
        elif characteristic == 0:
            assert raises(lambda: x / 2, DomainError)
            assert raises(lambda: x / two, DomainError)
            assert raises(lambda: two ** -1, DomainError)
            assert raises(lambda: two ** Z(-1), DomainError)
        else:
            # Non-prime modulus...
            # nmod can crash and fmpz_mod_poly won't crash but has awkward
            # behaviour under division.
            pass


def _all_matrices():
    """Return a list of matrix types and scalar types."""
    R163 = flint.fmpz_mod_ctx(163)
    R127 = flint.fmpz_mod_ctx(2**127 - 1)
    R255 = flint.fmpz_mod_ctx(2**255 - 19)
    return [
        # (matrix_type, scalar_type, is_field)
        (flint.fmpz_mat, flint.fmpz, False),
        (flint.fmpq_mat, flint.fmpq, True),
        (lambda *a: flint.nmod_mat(*a, 17), lambda x: flint.nmod(x, 17), True),
        (lambda *a: flint.fmpz_mod_mat(*a, R163), lambda x: flint.fmpz_mod(x, R163), True),
        (lambda *a: flint.fmpz_mod_mat(*a, R127), lambda x: flint.fmpz_mod(x, R127), True),
        (lambda *a: flint.fmpz_mod_mat(*a, R255), lambda x: flint.fmpz_mod(x, R255), True),
    ]


def _coercible_matrix_types(mat_type):
    """Return a list of matrix types that can be coerced to `mat_type`."""
    M = mat_type([[1, 2], [3, 4]])

    if type(M) is flint.fmpz_mat:
        return [
            flint.fmpz_mat
        ]
    elif type(M) is flint.fmpq_mat:
        return [
            flint.fmpz_mat,
            flint.fmpq_mat
        ]
    elif type(M) is flint.nmod_mat:
        m = M.modulus()
        return [
            flint.fmpz_mat,
            lambda *a: flint.nmod_mat(*a, m)
        ]
    elif type(M) is flint.fmpz_mod_mat:
        m = M.modulus()
        c = flint.fmpz_mod_ctx(m)
        mat_types = [
            flint.fmpz_mat,
            lambda *a: flint.fmpz_mod_mat(*a, c)
        ]
        if m < 2**64 - 1:
            mat_types.append(lambda *a: flint.nmod_mat(*a, m))
        return mat_types
    else:
        assert False


def _compatible_matrix_types(mat_type):
    """Return a list of matrix types that can be added to `mat_type`."""
    M = mat_type([[1, 2], [3, 4]])

    if type(M) is flint.fmpz_mat:
        return [
            flint.fmpz_mat
        ]
    elif type(M) is flint.fmpq_mat:
        return [
            # flint.fmpz_mat,
            flint.fmpq_mat
        ]
    elif type(M) is flint.nmod_mat:
        m = M.modulus()
        return [
            flint.fmpz_mat,
            lambda *a: flint.nmod_mat(*a, m)
        ]
    elif type(M) is flint.fmpz_mod_mat:
        m = M.modulus()
        c = flint.fmpz_mod_ctx(m)
        mat_types = [
            flint.fmpz_mat,
            lambda *a: flint.fmpz_mod_mat(*a, c)
        ]
        if m < 2**64 - 1:
            mat_types.append(lambda *a: flint.nmod_mat(*a, m))
        return mat_types
    else:
        assert False


def _incompatible_matrix_types(mat_type):
    M = mat_type([[1, 2], [3, 4]])

    if type(M) is flint.fmpz_mat:
        return []
    elif type(M) is flint.fmpq_mat:
        return [
            lambda *a: flint.nmod_mat(*a, 17),
            lambda *a: flint.fmpz_mod_mat(*a, flint.fmpz_mod_ctx(163)),
        ]
    elif type(M) is flint.nmod_mat:
        m = M.modulus()
        m2 = m - 1
        return [
            flint.fmpq_mat,
            lambda *a: flint.nmod_mat(*a, m2),
        ]
    elif type(M) is flint.fmpz_mod_mat:
        m = M.modulus()
        m2 = m - 1
        mat_types = [
            flint.fmpq_mat,
            lambda *a: flint.fmpz_mod_mat(*a, flint.fmpz_mod_ctx(m2)),
        ]
        if m < 2**64 - 1:
            mat_types.append(lambda *a: flint.nmod_mat(*a, m2))
        return mat_types
    else:
        assert False


def _poly_type_from_matrix_type(mat_type):
    M = mat_type([[1, 2], [3, 4]])

    if type(M) is flint.fmpz_mat:
        return flint.fmpz_poly
    elif type(M) is flint.fmpq_mat:
        return flint.fmpq_poly
    elif type(M) is flint.nmod_mat:
        m = M.modulus()
        return lambda *a: flint.nmod_poly(*a, m)
    elif type(M) is flint.fmpz_mod_mat:
        m = M.modulus()
        ctx = flint.fmpz_mod_ctx(m)
        pctx = flint.fmpz_mod_poly_ctx(ctx)
        return lambda *a: flint.fmpz_mod_poly(*a, pctx)
    else:
        assert False


def test_matrices_eq():
    for M, _, _ in _all_matrices():
        A1 = M([[1, 2], [3, 4]])
        A2 = M([[1, 2], [3, 4]])
        B = M([[5, 6], [7, 8]])
        assert (A1 == A2) is True
        assert (A1 != A2) is False
        assert (A1 == B) is False
        assert (A1 != B) is True
        assert (A1 == None) is False
        assert (A1 != None) is True
        assert (None == A1) is False
        assert (None != A1) is True
        assert raises(lambda: A1 < A2, TypeError)
        assert raises(lambda: A1 <= A2, TypeError)
        assert raises(lambda: A1 > A2, TypeError)
        assert raises(lambda: A1 >= A2, TypeError)

        for M2 in _incompatible_matrix_types(M):
            A1 = M([[1, 2], [3, 4]])
            A2 = M2([[1, 2], [3, 4]])
            assert (A1 == A2) is False
            assert (A1 != A2) is True


def test_matrices_constructor():
    for M, S, _ in _all_matrices():
        assert raises(lambda: M(), TypeError)

        # Empty matrices
        assert M([]).nrows() == 0
        assert M([]).ncols() == 0
        assert M([]) != M([[]])
        assert M([[]]).nrows() == 1
        assert M([[]]).ncols() == 0
        assert M([]) == M(0, 0, []) == M(0, 0)
        assert M([[]]) == M(1, 0, []) == M(1, 0)
        assert M([[], []]) == M(2, 0, []) == M(2, 0)
        assert M(0, 2, []) == M(0, 2) != M(2, 0)

        # Basic construction
        M1234 = M([[1, 2], [3, 4]])
        assert M1234.nrows() == 2
        assert M1234.ncols() == 2
        assert M1234 == M([[1, 2], [3, 4]])
        assert M1234 != M([[1, 2], [3, 5]])
        assert M1234 == M(2, 2, [1, 2, 3, 4])
        assert M1234.entries() == [S(1), S(2), S(3), S(4)]
        assert M1234.table() == [[S(1), S(2)], [S(3), S(4)]]
        assert M1234[0, 0] == S(1)
        assert M1234[0, 1] == S(2)
        assert M1234[1, 0] == S(3)
        assert M1234[1, 1] == S(4)
        assert type(M1234[0, 0]) is type(S(1))
        assert M1234 == M([[S(1), S(2)], [S(3), S(4)]])
        assert M1234 == M(2, 2, [S(1), S(2), S(3), S(4)])

        # Non-square matrices
        M6 = M([[1, 2, 3], [4, 5, 6]])
        assert M6.nrows() == 2
        assert M6.ncols() == 3
        assert M6 == M(2, 3, [1, 2, 3, 4, 5, 6])
        assert M6.entries() == [S(1), S(2), S(3), S(4), S(5), S(6)]
        assert M6 != M1234

        # Bad list inputs
        assert raises(lambda: M(None), TypeError)
        assert raises(lambda: M([None]), TypeError)
        assert raises(lambda: M(2, 3, [1, 2, 3, 4, 5]), ValueError)
        assert raises(lambda: M([[1, 2], [3, 4], [5, 6, 7]]), ValueError)
        assert raises(lambda: M([[1, 2], [3, 4], [5, 6, 7, 8]]), ValueError)
        assert raises(lambda: M([[1, 2], [3, None]]), TypeError)

        # Construct from a coercible matrix
        M1234 = M([[1, 2], [3, 4]])
        for M2 in _coercible_matrix_types(M):
            if type(M2([[1]])) is flint.nmod_mat:
                continue
            M1234_2 = M(M2([[1, 2], [3, 4]]))
            assert M1234_2 == M1234
            assert type(M1234_2) is type(M1234)


def _matrix_repr(M):
    item = M[0, 0]
    if type(item) is flint.fmpz:
        return f"fmpz_mat({M.nrows()}, {M.ncols()}, {M.entries()!r})"
    elif type(item) is flint.fmpq:
        return f"fmpq_mat({M.nrows()}, {M.ncols()}, {M.entries()!r})"
    elif type(item) is flint.nmod:
        return f"nmod_mat({M.nrows()}, {M.ncols()}, {M.entries()!r}, {M.modulus()})"
    elif type(item) is flint.fmpz_mod:
        return f"fmpz_mod_mat({M.nrows()}, {M.ncols()}, {M.entries()!r}, {M.modulus()})"
    else:
        assert False


def test_matrices_strrepr():
    for M, _, _ in _all_matrices():
        A = M([[1, 2], [3, 4]])
        A_str = "[1, 2]\n[3, 4]"
        A_repr = _matrix_repr(A)

        assert A.str() == A_str, type(A).__name__
        assert A.repr() == A_repr, type(A).__name__

        # str always returns a pretty result
        assert str(A) == A_str, type(A).__name__

        # repr depends on ctx.pretty
        pretty = ctx.pretty
        try:
            ctx.pretty = True
            assert repr(A) == A_str, type(A).__name__
            ctx.pretty = False
            assert repr(A) == A_repr, type(A).__name__
        finally:
            ctx.pretty = pretty


def test_matrices_getitem():
    for M, S, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        assert M1234[0, 0] == S(1)
        assert M1234[0, 1] == S(2)
        assert M1234[1, 0] == S(3)
        assert M1234[1, 1] == S(4)
        assert raises(lambda: M1234[0, 2], IndexError)
        assert raises(lambda: M1234[2, 0], IndexError)
        assert raises(lambda: M1234[2, 2], IndexError)
        # XXX: Should negative indices be allowed?
        assert raises(lambda: M1234[-1, 0], IndexError)
        assert raises(lambda: M1234[0, -1], IndexError)
        assert raises(lambda: M1234[-1, -1], IndexError)


def test_matrices_setitem():
    for M, S, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])

        assert M1234[0, 0] == S(1)
        M1234[0, 0] = S(5)
        assert M1234[0, 0] == S(5)
        assert M1234.table() == [[S(5), S(2)], [S(3), S(4)]]

        def setbad(obj, key, val):
            obj[key] = val

        assert raises(lambda: setbad(M1234, (0,0), None), TypeError)
        assert raises(lambda: setbad(M1234, (0,None), 1), TypeError)
        assert raises(lambda: setbad(M1234, (None,0), 1), TypeError)
        assert raises(lambda: setbad(M1234, None, 1), TypeError)

        assert raises(lambda: setbad(M1234, (0,2), 1), IndexError)
        assert raises(lambda: setbad(M1234, (2,0), 1), IndexError)
        assert raises(lambda: setbad(M1234, (2,2), 1), IndexError)
        # XXX: Should negative indices be allowed?
        assert raises(lambda: setbad(M1234, (-1,0), 1), IndexError)
        assert raises(lambda: setbad(M1234, (0,-1), 1), IndexError)
        assert raises(lambda: setbad(M1234, (-1,-1), 1), IndexError)


def test_matrices_bool():
    for M, _, _ in _all_matrices():
        assert bool(M([])) is False
        assert bool(M([[0]])) is False
        assert bool(M([[1]])) is True
        assert bool(M([[0, 0], [0, 0]])) is False
        assert bool(M([[1, 0], [0, 0]])) is True
        assert bool(M([[0, 0], [0, 1]])) is True
        assert bool(M([[1, 0], [0, 1]])) is True


def test_matrices_pos_neg():
    for M, _, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        assert +M1234 == M1234
        assert -M1234 == M([[-1, -2], [-3, -4]])


def test_matrices_add():
    for M, _, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        M5678 = M([[5, 6], [7, 8]])
        assert M1234 + M5678 == M([[6, 8], [10, 12]])
        assert raises(lambda: M1234 + 1, TypeError)
        assert raises(lambda: 1 + M1234, TypeError)
        assert raises(lambda: M1234 + None, TypeError)
        assert raises(lambda: None + M1234, TypeError)
        assert raises(lambda: M1234 + M([[1, 2, 3], [4, 5, 6]]), ValueError)
        for M2 in _compatible_matrix_types(M):
            assert M1234 + M2([[1, 2], [3, 4]]) == M([[2, 4], [6, 8]])
            assert M2([[1, 2], [3, 4]]) + M1234 == M([[2, 4], [6, 8]])
            assert raises(lambda: M1234 + M2([[1, 2, 3], [4, 5, 6]]), ValueError)
            assert raises(lambda: M2([[1, 2, 3], [4, 5, 6]]) + M1234, ValueError)
        for M2 in _incompatible_matrix_types(M):
            assert raises(lambda: M1234 + M2([[1, 2], [3, 4]]), (TypeError, ValueError))
            assert raises(lambda: M2([[1, 2], [3, 4]]) + M1234, (TypeError, ValueError))


def test_matrices_sub():
    for M, _, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        M5678 = M([[5, 6], [7, 8]])
        assert M1234 - M5678 == M([[-4, -4], [-4, -4]])
        assert raises(lambda: M1234 - 1, TypeError)
        assert raises(lambda: 1 - M1234, TypeError)
        assert raises(lambda: M1234 - None, TypeError)
        assert raises(lambda: None - M1234, TypeError)
        assert raises(lambda: M1234 - M([[1, 2, 3], [4, 5, 6]]), ValueError)
        for M2 in _compatible_matrix_types(M):
            assert M1234 - M2([[1, 2], [3, 4]]) == M([[0, 0], [0, 0]])
            assert M2([[1, 2], [3, 4]]) - M1234 == M([[0, 0], [0, 0]])
            assert raises(lambda: M1234 - M2([[1, 2, 3], [4, 5, 6]]), ValueError)
            assert raises(lambda: M2([[1, 2, 3], [4, 5, 6]]) - M1234, ValueError)
        for M2 in _incompatible_matrix_types(M):
            assert raises(lambda: M1234 - M2([[1, 2], [3, 4]]), (TypeError, ValueError))
            assert raises(lambda: M2([[1, 2], [3, 4]]) - M1234, (TypeError, ValueError))


def test_matrices_mul():
    for M, S, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        M5678 = M([[5, 6], [7, 8]])
        assert M1234 * M5678 == M([[19, 22], [43, 50]])
        M21 = M([[2], [1]])
        assert M1234 * M21 == M([[4], [10]])

        assert 2 * M1234 == M([[2, 4], [6, 8]])
        assert M1234 * 2 == M([[2, 4], [6, 8]])
        assert S(2) * M1234 == M([[2, 4], [6, 8]])
        assert M1234 * S(2) == M([[2, 4], [6, 8]])
        assert raises(lambda: M1234 * None, TypeError)
        assert raises(lambda: None * M1234, TypeError)
        assert raises(lambda: M1234 * M([[1, 2], [3, 4], [5, 6]]), ValueError)
        assert raises(lambda: M([[1, 2, 3], [4, 5, 6]]) * M1234, ValueError)

        for M2 in _compatible_matrix_types(M):
            assert M1234 * M2([[1, 2], [3, 4]]) == M([[7, 10], [15, 22]])
            assert M2([[1, 2], [3, 4]]) * M1234 == M([[7, 10], [15, 22]])

        for M2 in _incompatible_matrix_types(M):
            assert raises(lambda: M1234 * M2([[1, 2], [3, 4]]), (TypeError, ValueError))
            assert raises(lambda: M2([[1, 2], [3, 4]]) * M1234, (TypeError, ValueError))


def test_matrices_pow():
    for M, _, is_field in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        assert M1234**0 == M([[1, 0], [0, 1]])
        assert M1234**1 == M1234
        assert M1234**2 == M([[7, 10], [15, 22]])
        assert M1234**3 == M([[37, 54], [81, 118]])
        if is_field:
            assert M1234**-1 == M([[-4, 2], [3, -1]]) / 2
            assert M1234**-2 == M([[22, -10], [-15, 7]]) / 4
            assert M1234**-3 == M([[-118, 54], [81, -37]]) / 8
            Ms = M([[1, 2], [3, 6]])
            assert raises(lambda: Ms**-1, ZeroDivisionError)
        Mr = M([[1, 2, 3], [4, 5, 6]])
        assert raises(lambda: Mr**0, ValueError)
        assert raises(lambda: Mr**1, ValueError)
        assert raises(lambda: Mr**2, ValueError)
        assert raises(lambda: M1234**None, TypeError)
        assert raises(lambda: None**M1234, TypeError)


def test_matrices_div():
    for M, S, is_field in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        if is_field:
            assert M1234 / 2 == M([[S(1)/2, S(1)], [S(3)/2, 2]])
            assert M1234 / S(2) == M([[S(1)/2, S(1)], [S(3)/2, 2]])
            assert raises(lambda: M1234 / 0, ZeroDivisionError)
            assert raises(lambda: M1234 / S(0), ZeroDivisionError)
        raises(lambda: M1234 / None, TypeError)
        raises(lambda: None / M1234, TypeError)


def test_matrices_properties():
    for M, _, _ in _all_matrices():
        # XXX: Add these properties to all matrix types
        if M is not flint.fmpz_mat:
            continue

        assert M([[1, 2], [3, 4]]).is_square() is True
        assert M([[1, 2]]).is_square() is False
        assert M(0, 2, []).is_square() is False
        assert M(2, 0, []).is_square() is False

        assert M([[1, 2], [3, 4]]).is_empty() is False
        assert M(0, 2, []).is_empty() is True
        assert M(2, 0, []).is_empty() is True

        assert M([[1, 2], [3, 4]]).is_zero() is False
        assert M([[0, 0], [0, 0]]).is_zero() is True

        assert M([[1, 0], [0, 1]]).is_one() is True
        assert M([[1, 2], [3, 4]]).is_one() is False
        assert M([[1, 0], [0, 1], [0, 0]]).is_one() is True # ??
        assert M(0, 0, []).is_one() is True

        assert M([[-1, 0], [0, -1]]).is_neg_one() is True
        assert M([[-1, 0], [0, 1]]).is_neg_one() is False
        assert M([[-1, -1], [-1, -1]]).is_neg_one() is False
        assert M([[-1, 0], [0, -1], [0, 0]]).is_neg_one() is False # ??
        assert M(0, 0, []).is_neg_one() is True

        assert M([[2, 0], [0, 2]]).is_scalar() is True
        assert M([[2, 0], [0, 3]]).is_scalar() is False
        assert M([[1, 0], [0, 1], [0, 0]]).is_scalar() is False

        assert M([[1, 0], [0, 2]]).is_diagonal() is True
        assert M([[1, 0], [1, 2]]).is_diagonal() is False
        assert M([[1, 0], [0, 1], [0, 0]]).is_diagonal() is True

        assert M([[1, 1, 1], [0, 2, 2]]).is_upper_triangular() is True
        assert M([[1, 1, 1], [1, 2, 2]]).is_upper_triangular() is False

        assert M([[1, 0, 0], [1, 2, 0]]).is_lower_triangular() is True
        assert M([[1, 1, 0], [1, 2, 0]]).is_lower_triangular() is False


def test_matrices_inv():
    for M, S, is_field in _all_matrices():
        if is_field:
            M1234 = M([[1, 2], [3, 4]])
            assert M1234.inv() == M([[-2, 1], [S(3)/2, -S(1)/2]])
            M1236 = M([[1, 2], [3, 6]])
            assert raises(lambda: M1236.inv(), ZeroDivisionError)
            Mr = M([[1, 2, 3], [4, 5, 6]])
            assert raises(lambda: Mr.inv(), ValueError)
        # XXX: Test non-field matrices. unimodular?


def test_matrices_det():
    for M, S, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        assert M1234.det() == S(-2)
        M9 = M([[1, 2, 3], [4, 5, 6], [7, 8, 10]])
        assert M9.det() == S(-3)
        Mr = M([[1, 2, 3], [4, 5, 6]])
        assert raises(lambda: Mr.det(), ValueError)


def test_matrices_charpoly():
    for M, _, _ in _all_matrices():
        P = _poly_type_from_matrix_type(M)
        M1234 = M([[1, 2], [3, 4]])
        assert M1234.charpoly() == P([-2, -5, 1])
        M9 = M([[1, 2, 3], [4, 5, 6], [7, 8, 10]])
        assert M9.charpoly() == P([3, -12, -16, 1])
        Mr = M([[1, 2, 3], [4, 5, 6]])
        assert raises(lambda: Mr.charpoly(), ValueError)


def test_matrices_minpoly():
    for M, _, _ in _all_matrices():
        P = _poly_type_from_matrix_type(M)
        M1234 = M([[1, 2], [3, 4]])
        assert M1234.minpoly() == P([-2, -5, 1])
        M9 = M([[2, 1, 0], [0, 2, 0], [0, 0, 2]])
        assert M9.minpoly() == P([4, -4, 1])
        Mr = M([[1, 2, 3], [4, 5, 6]])
        assert raises(lambda: Mr.minpoly(), ValueError)


def test_matrices_rank():
    for M, _, _ in _all_matrices():
        M1234 = M([[1, 2], [3, 4]])
        assert M1234.rank() == 2
        Mr = M([[1, 2, 3], [4, 5, 6]])
        assert Mr.rank() == 2
        Ms = M([[1, 2], [2, 4]])
        assert Ms.rank() == 1
        Mz = M([[0, 0], [0, 0]])
        assert Mz.rank() == 0


def test_matrices_rref():
    for M, _, is_field in _all_matrices():
        if is_field:
            Mr = M([[1, 2, 3], [4, 5, 6]])
            Mr_rref = M([[1, 0, -1], [0, 1, 2]])
            assert Mr.rref() == (Mr_rref, 2)
            assert Mr == M([[1, 2, 3], [4, 5, 6]])
            assert Mr.rref(inplace=True) == (Mr_rref, 2)
            assert Mr == Mr_rref


def test_matrices_fflu():

    QQ = flint.fmpq_mat
    shape = lambda A: (A.nrows(), A.ncols())

    def is_permutation(P):
        if not P.is_square():
            return False
        n = P.nrows()
        for i, row in enumerate(sorted(P.tolist(), reverse=True)):
            if row != [int(i == j) for j in range(n)]:
                return False
        return True

    def check_fflu(A):
        m, n = shape(A)
        P, L, D, U = A.fflu()
        Dq = QQ(D)
        assert P*A == L*Dq.inv()*U
        assert shape(P) == shape(L) == shape(D) == (m, m)
        assert shape(A) == shape(U) == (m, n)
        assert is_permutation(P)
        assert L.is_lower_triangular()
        assert U.is_upper_triangular()
        assert D.is_diagonal()

    for M, _, _ in _all_matrices():
        # XXX: Add this to more matrix types...
        if M is not flint.fmpz_mat:
            continue

        A = M([[1, 2], [3, 4]])
        P, L, D, U = A.fflu()
        assert P == M([[1, 0], [0, 1]])
        assert L == M([[1, 0], [3, -2]])
        assert D == M([[1, 0], [0, -2]])
        assert U == M([[1, 2], [0, -2]])

        check_fflu(M(0, 0, []))
        check_fflu(M(2, 0, []))
        check_fflu(M(0, 2, []))
        check_fflu(M([[1]]))

        check_fflu(M([[1, 2], [3, 4]]))
        check_fflu(M([[1, 2, 3], [4, 5, 6]]))
        check_fflu(M([[1, 2], [3, 4], [5, 6]]))
        check_fflu(M([[1, 2], [2, 4]]))
        check_fflu(M([[0, 0], [0, 0]]))
        check_fflu(M([[1, 1, 1], [1, 1, 1]]))

        for _ in range(10):
            for m in range(1, 5):
                for n in range(1, 5):
                    A = M.randbits(m, n, 10)
                    check_fflu(A)


def test_matrices_solve():
    for M, _, is_field in _all_matrices():
        if is_field:
            A = M([[1, 2], [3, 4]])
            x = M([[1], [2]])
            b = M([[5], [11]])
            assert A*x == b
            assert A.solve(b) == x
            A22 = M([[1, 2], [3, 4]])
            A23 = M([[1, 2, 3], [4, 5, 6]])
            b2 = M([[5], [11]])
            b3 = M([[5], [11], [17]])
            assert raises(lambda: A22.solve(b3), ValueError)
            assert raises(lambda: A23.solve(b2), ValueError)
            assert raises(lambda: A.solve(None), TypeError)
            A = M([[1, 2], [2, 4]])
            assert raises(lambda: A.solve(b), ZeroDivisionError)


def test_matrices_transpose():
    for M, _, _ in _all_matrices():
        M1234 = M([[1, 2, 3], [4, 5, 6]])
        assert M1234.transpose() == M([[1, 4], [2, 5], [3, 6]])


def test_fq_default():
    # test fq_default context creation

    # fq_type parsing
    assert raises(lambda: flint.fq_default_ctx(5, fq_type="A"), ValueError)
    assert raises(lambda: flint.fq_default_ctx(5, fq_type=[]), TypeError)
    assert raises(lambda: flint.fq_default_ctx(5, fq_type=-1), ValueError)
    assert raises(lambda: flint.fq_default_ctx("ABC"), TypeError) # type: ignore

    # var must be one character
    assert raises(lambda: flint.fq_default_ctx(5, var="XXX"), ValueError)

    # p must be set if modulus has no characteristic / modulus
    assert raises(lambda: flint.fq_default_ctx(modulus=[0,1,0]), ValueError)

    # prime must be prime when setting from modulus
    assert raises(lambda: flint.fq_default_ctx(10, modulus=[0,1,0]), ValueError)
    mod_not_prime = flint.fmpz_mod_poly_ctx(10)([1,0,1])
    assert raises(lambda: flint.fq_default_ctx(modulus=mod_not_prime), ValueError)
    mod_not_irr = flint.fmpz_mod_poly_ctx(11)([0,0,1])
    assert raises(lambda: flint.fq_default_ctx(modulus=mod_not_irr), ValueError)

    # modulus must be able to be cast to fmpz_mod_poly
    assert raises(lambda: flint.fq_default_ctx(11, modulus="AAA"), TypeError)

    # either p or modulus must be set
    assert raises(lambda: flint.fq_default_ctx(p=None, modulus=None), ValueError)

    # p must be prime
    assert raises(lambda: flint.fq_default_ctx(10), ValueError)

    # degree must be positive
    assert raises(lambda: flint.fq_default_ctx(11, -1), ValueError)

    # GF(5)
    gf_5 = flint.fq_default_ctx(5, fq_type='NMOD')
    gf_5_ = flint.fq_default_ctx(5, fq_type='NMOD')

    # GF(5^2)
    gf_5_2 = flint.fq_default_ctx(5, 2, fq_type='FQ_ZECH')
    gf_5_2_ = flint.fq_default_ctx(5, 2, fq_type='FQ_ZECH')

    # GF((2**127 - 1)^2)
    gf_127 = flint.fq_default_ctx(2**127 - 1, 2)
    gf_127_2 = flint.fq_default_ctx(2**127 - 1, 2)

    assert (gf_5 == gf_5_) is True
    assert (hash(gf_5) == hash(gf_5_)) is True
    assert (gf_5 != gf_5_) is False
    assert (gf_5 == gf_5_2) is False
    assert (gf_5 != gf_5_2) is True
    assert (gf_5 == "a") is False
    assert (gf_5 != "a") is True

    assert gf_5.prime() == gf_5_2.prime() == 5
    assert gf_5_2.order() == 5*5
    assert gf_5_2.multiplicative_order() == 5*5 - 1
    assert gf_127_2.prime() == 2**127 - 1

    assert gf_5_2(0) == gf_5_2.zero()
    assert gf_5_2(1) == gf_5_2.one()
    assert gf_5_2.gen() == gf_5_2([0, 1])

    assert str(gf_5) == "Context for fq_default in GF(5)"
    assert str(gf_5_2) == "Context for fq_default in GF(5^2)[z]/(z^2 + 4*z + 2)"

    assert repr(gf_5) == "fq_default_ctx(5, var='z' type='NMOD')"
    assert repr(gf_5_2) == "fq_default_ctx(5, 2, 'z', x^2 + 4*x + 2, 'FQ_ZECH')"

    # coercision
    assert gf_5.one() == flint.fq_default(1, gf_5)
    assert gf_5(1) == gf_5.one()
    assert gf_5(flint.fmpz(1)) == gf_5.one()
    assert gf_5(-1) == -gf_5.one()
    assert gf_5(flint.fmpz(-1)) == -gf_5.one()
    R = flint.fmpz_mod_ctx(5)
    assert gf_5(R(1)) == gf_5.one()
    assert gf_5(R(-1)) == -gf_5.one()
    assert gf_5(flint.nmod(1, 5)) == gf_5.one()
    assert gf_5(flint.nmod(-1, 5)) == -gf_5.one()
    assert gf_5([0, 1]) == gf_5.gen()
    assert gf_5(flint.fmpz_poly([0, 1])) == gf_5.gen()
    R = flint.fmpz_mod_poly_ctx(5)
    assert gf_5.gen() == gf_5(R.gen())
    assert gf_5.gen() == gf_5(flint.nmod_poly([0, 1], 5))
    assert gf_5(flint.fmpz(2**64)) == gf_5(2**64)
    assert raises(lambda: flint.fq_default(1, "AAA"), TypeError) # type: ignore
    assert raises(lambda: flint.fq_default.__init__(1, "AAA"), TypeError) # type: ignore
    assert raises(lambda: flint.fq_default("AAA", gf_5), TypeError)
    assert raises(lambda: gf_5.one() + gf_5_2.one(), ValueError)
    # testing various equalties between types

    # integers are the same if characteristic is the same
    # even with extensions
    assert gf_5.one() == gf_5_.one()
    assert hash(gf_5.one()) == hash(gf_5_.one())
    assert gf_5.one() == gf_5_2.one()
    assert gf_5.one() != gf_127.one()

    # the generators for different extensions
    assert gf_5_2([0, 1]) != gf_5([0, 1])
    assert gf_5_2([0, 1]) == gf_5_2_([0, 1])
    assert gf_5_2([0, 1]) != gf_127_2([0, 1])

    # integers are reduced modulo p before comparison
    for int_type in [int, flint.fmpz]:
        assert gf_5(1) == int_type(1)
        assert gf_5(-1) == int_type(-1)
        assert gf_5(-1) == int_type(4)
        assert gf_5(4) == int_type(4)
        assert gf_5(4) == int_type(-1)

    # integers modulo n also can be compared when they match
    assert gf_5(1) == flint.nmod(1, 5)
    assert gf_5(-1) == flint.nmod(-1, 5)
    assert gf_5(-1) == flint.nmod(4, 5)
    assert gf_5_2(1) == flint.nmod(1, 5)
    assert gf_5_2(-1) == flint.nmod(-1, 5)
    assert gf_5_2(-1) == flint.nmod(4, 5)

    # when the moduli dont match, comparison is always false
    assert gf_5(1) != flint.nmod(1, 7)
    assert gf_5(-1) != flint.nmod(-1, 7)
    assert gf_5(-1) != flint.nmod(4, 7)
    assert gf_5_2(1) != flint.nmod(1, 7)
    assert gf_5_2(-1) != flint.nmod(-1, 7)
    assert gf_5_2(-1) != flint.nmod(4, 7)

    # integers modulo n also can be compared when they match
    R5 = flint.fmpz_mod_ctx(5)
    assert gf_5(1) == R5(1)
    assert gf_5(-1) == R5(-1)
    assert gf_5(-1) == R5(4)
    assert gf_5_2(1) == R5(1)
    assert gf_5_2(-1) == R5(-1)
    assert gf_5_2(-1) == R5(4)

    # when the moduli dont match, comparison is always false
    R7 = flint.fmpz_mod_ctx(7)
    assert gf_5(1) != R7(1)
    assert gf_5(-1) != R7(-1)
    assert gf_5(-1) != R7(4)
    assert gf_5_2(1) != R7(1)
    assert gf_5_2(-1) != R7(-1)
    assert gf_5_2(-1) != R7(4)

    # test fq_default element arithmetic

    for gf in [gf_5, gf_5_2, gf_127, gf_127_2]:

        assert (gf(0) == gf.zero()) is True
        assert (gf(0) != gf.zero()) is False
        assert (gf(1) == gf.zero()) is False
        assert (gf(1) != gf.zero()) is True
        assert raises(lambda: gf.zero() > gf.zero(), TypeError) # type: ignore
        assert raises(lambda: gf.zero() >= gf.zero(), TypeError) # type: ignore
        assert raises(lambda: gf.zero() < gf.zero(), TypeError) # type: ignore
        assert raises(lambda: gf.zero() <= gf.zero(), TypeError) # type: ignore

        assert gf.zero().is_zero() is True
        assert gf.one().is_zero() is False

        assert gf.zero().is_one() is False
        assert gf.one().is_one() is True

        a = gf.random_element(not_zero=True)
        b = gf.random_element(not_zero=True)
        c = gf.random_element(not_zero=True)

        assert a + (-a) == gf.zero()
        assert a + a == 2*a
        assert a * a == a**2
        assert a * a == a.square()
        assert a * a * a == pow(a, 3)

        assert (a + b) + c == a + (b + c)
        assert (a - b) - c == a - (b + c)
        assert (a * b) * c == a * (b * c)
        assert (a / b) / c == a / (b * c)

        assert a + 0 == 0 + a == a
        assert a - 0 == -(0 - a) == a
        assert a + gf.zero() == a
        assert a * 1 == 1 * a == a
        assert a * gf.one() == a
        assert a * gf.zero() == gf.zero()
        assert a / a == gf.one()

        assert raises(lambda: a / 0, ZeroDivisionError)
        assert raises(lambda: ~gf.zero(), ZeroDivisionError)
        assert raises(lambda: pow(gf.zero(), -1), ZeroDivisionError)
        assert raises(lambda: pow(gf.zero(), "A"), TypeError) # type: ignore

        assert 1/a == pow(a, -1) == ~a
        assert gf.one() == pow(a, 0)
        assert gf.zero() == pow(gf.zero(), 2**64)
        assert a == pow(a, 1)
        assert pow(a, flint.fmpz(2**64)) == pow(a, 2**64)
        assert (a*a).is_square()
        assert (a*a).sqrt() in [a, -a]

        while True:
            nqr = gf.random_element()
            if not nqr.is_square():
                break
        assert raises(lambda: nqr.sqrt(), DomainError)


def test_fq_default_poly():
    F = flint.fq_default_ctx(11, 3)
    R1 = flint.fq_default_poly_ctx(F)
    R2 = flint.fq_default_poly_ctx(11, 3)
    R3 = flint.fq_default_poly_ctx(13, 5)

    assert raises(lambda: flint.fq_default_poly_ctx("AAA"), TypeError) # type: ignore
    assert (R1 == R1) is True
    assert hash(R1) == hash(R2)
    assert (R1 != R1) is False
    assert (R1 == R2) is True
    assert (R1 != R2) is False
    assert (R1 != R3) is True
    assert (R1 == R3) is False
    assert (R1 != "AAA") is True
    assert (R1 == "AAA") is False

    assert str(R1) == "Context for fq_default_poly with field: Context for fq_default in GF(11^3)[z]/(z^3 + 2*z + 9)"
    assert str(R1) == str(R2)
    assert repr(R3) == "fq_default_poly_ctx(fq_default_ctx(13, 5, 'z', x^5 + 4*x + 11, 'FQ_NMOD'))"

    # random element failure
    f = R1.random_element(not_zero=True)
    assert not f.is_zero()
    assert raises(lambda: R1.random_element(monic="AAA"), TypeError) # type: ignore
    assert raises(lambda: R1.random_element(degree=-1), ValueError)

    assert raises(lambda: flint.fq_default_poly([1,2,3], "AAA"), TypeError) # type: ignore

    assert R1(0).leading_coefficient() == 0
    assert raises(lambda: R1.random_element().reverse(degree=-1), ValueError)

    # some coercion
    assert raises(lambda: R3(F(1)), ValueError)
    assert R1.one() == R1(1)
    assert R1.one() == R1([1])
    assert R1.one() == R1(flint.fmpz(1))
    assert R1.one() == R1(flint.fmpz_poly([1]))
    assert R1.one() == R1(flint.fmpz_mod_ctx(11)(1))
    assert R1.one() == R1(flint.fmpz_mod_poly_ctx(11)(1))
    assert R1.one() == R1(flint.nmod_poly(1, 11))

    R_sml = flint.fq_default_poly_ctx(5)
    R_med = flint.fq_default_poly_ctx(65537)
    R_big = flint.fq_default_poly_ctx(2**127 - 1)
    R_sml_ext = flint.fq_default_poly_ctx(5, 5)
    R_med_ext = flint.fq_default_poly_ctx(65537, 3)
    R_big_ext = flint.fq_default_poly_ctx(2**127 - 1, 2)

    F_cmp = flint.fq_default_ctx(11)
    R_cmp = flint.fq_default_poly_ctx(F_cmp)
    f_cmp = R_cmp([1,2,3,4,5])

    for R_test in [R_sml, R_med, R_big, R_sml_ext, R_med_ext, R_big_ext]:
        F_test = R_test.base_field()
        while True:
            nqr = F_test.random_element()
            if not nqr.is_square():
                break

        f = R_test([-1,-2])
        g = R_test([-3,-4])
        assert (f == f) is True
        assert (f != g) is True
        assert (hash(f) == hash(f)) is True
        assert (hash(f) != hash(g)) is True

        # Exact division
        assert raises(lambda: f.exact_division(f_cmp), ValueError)
        assert raises(lambda: f.exact_division("AAA"), TypeError)
        assert raises(lambda: f.exact_division(0), ZeroDivisionError)
        assert (f * g).exact_division(g) == f
        assert raises(lambda: f.exact_division(g), DomainError)
        assert raises(lambda: f / "AAA", TypeError)
        assert raises(lambda: "AAA" / f, TypeError)

        # ZeroDivisionError
        assert raises(lambda: f / 0, ZeroDivisionError)
        assert raises(lambda: f // 0, ZeroDivisionError)
        assert raises(lambda: 1 / R_test.zero(), ZeroDivisionError)
        assert raises(lambda: 1 // R_test.zero(), ZeroDivisionError)
        assert raises(lambda: 1 % R_test.zero(), ZeroDivisionError)

        # pow
        # assert ui and fmpz exp agree for polynomials and generators
        R_gen = R_test.gen()
        assert raises(lambda: f**(-2), DomainError)
        assert pow(f, 2**60, g) == pow(pow(f, 2**30, g), 2**30, g)
        assert pow(R_gen, 2**60, g) == pow(pow(R_gen, 2**30, g), 2**30, g)
        assert raises(lambda: pow(f, -2, g), ValueError)
        assert raises(lambda: pow(f, 1, "A"), TypeError)
        assert raises(lambda: pow(f, "A", g), TypeError) # type: ignore
        assert raises(lambda: f.pow_mod(2**32, g, mod_rev_inv="A"), TypeError) # type: ignore

        # Shifts
        assert raises(lambda: R_test([1,2,3]).left_shift(-1), ValueError)
        assert raises(lambda: R_test([1,2,3]).right_shift(-1), ValueError)
        assert R_test([1,2,3]).left_shift(3) == R_test([0,0,0,1,2,3])
        assert R_test([1,2,3]).right_shift(1) == R_test([2,3])

        # mulmod
        assert f.mul_mod(f, g) == (f*f) % g
        assert raises(lambda: f.mul_mod(f, "AAA"), TypeError)
        assert raises(lambda: f.mul_mod("AAA", g), TypeError)

        # pow_mod
        assert f.pow_mod(2, g) == (f*f) % g
        assert raises(lambda: f.pow_mod(2, "AAA"), TypeError)

        # roots
        assert raises(lambda: f.real_roots(), DomainError)
        assert raises(lambda: f.complex_roots(), DomainError)

        # compose errors
        assert raises(lambda: f.compose("A"), TypeError)
        assert raises(lambda: f.compose_mod("A", g), TypeError)
        assert raises(lambda: f.compose_mod(g, "A"), TypeError)
        assert raises(lambda: f.compose_mod(g, R_test.zero()), ZeroDivisionError)

        # inverse_mod
        while True:
            # Ensure f is invertible
            f = R_test.random_element()
            if not f.constant_coefficient().is_zero():
                break
        while True:
            h = R_test.random_element()
            if f.gcd(h).is_one():
                break
        g = f.inverse_mod(h)
        assert f.mul_mod(g, h).is_one()
        if f.degree() >= 1:
            assert raises(lambda: f.inverse_mod(2*f), ValueError)
        else:
            assert f.inverse_mod(2*f) == 0 # ???

        # series
        f_non_square = R_test([nqr, 1, 1, 1])
        f_zero = R_test([0, 1, 1, 1])
        assert raises(lambda: f_non_square.sqrt_trunc(1), ValueError)
        assert raises(lambda: f_zero.sqrt_trunc(1), ZeroDivisionError)
        assert raises(lambda: f_non_square.inv_sqrt_trunc(1), ValueError)
        assert raises(lambda: f_zero.inv_sqrt_trunc(1), ZeroDivisionError)
        f_inv = f.inverse_series_trunc(2)
        assert (f * f_inv) % R_test([0,0,1]) == 1
        assert raises(lambda: R_test([0,1]).inverse_series_trunc(2), ZeroDivisionError)

        # deflation
        f1 = R_test([1,0,2,0,3])
        assert raises(lambda: f1.deflate(100), ValueError)
        assert f1.deflate(2) == R_test([1,2,3])

        # truncate things
        f = R_test.random_element()
        g = R_test.random_element()
        h = R_test.random_element()
        x = R_test.gen()
        f_trunc = f % x**3

        assert f.equal_trunc(f_trunc, 3)
        assert not f.equal_trunc("A", 3)
        assert not f.equal_trunc(f_cmp, 3)

        assert raises(lambda: f.add_trunc("A", 1), TypeError)
        assert raises(lambda: f.add_trunc(f_cmp, 1), ValueError)
        assert f.add_trunc(g, 3) == (f + g) % x**3

        assert raises(lambda: f.sub_trunc("A", 1), TypeError)
        assert raises(lambda: f.sub_trunc(f_cmp, 1), ValueError)
        assert f.sub_trunc(g, 3) == (f - g) % x**3

        assert raises(lambda: f.mul_low("A", 1), TypeError)
        assert raises(lambda: f.mul_low(g, "A"), TypeError) # type: ignore
        assert raises(lambda: f.mul_low(f_cmp, 1), ValueError)
        assert f.mul_low(g, 3) == (f * g) % x**3

        assert raises(lambda: f.pow_trunc(-1, 5), ValueError)


def test_python_threads():
    #
    # https://github.com/flintlib/python-flint/issues/224
    #
    # XXX: This test crashes under the free-threading mode because of memory
    # corruption.
    #
    # It is not clear if this should be fixed or if mutating
    # matrices/polynomials that are shared between multiple threads should just
    # be disallowed.
    #
    # This thread is skipped on Emscripten/WASM builds as we can't start new
    # threads in Pyodide.

    # Skip the test on the free-threaded build and on WASM...
    import sys
    if (sys.version_info[:2] >= (3, 13) and not sys._is_gil_enabled()) or ( # type: ignore
        sys.platform == "emscripten" or platform.machine() in ["wasm32", "wasm64"]):
        return

    from threading import Thread

    iterations = 10**5
    nthreads = 3 + 1
    size = 3
    M = flint.fmpz_mat([[0]*size for _ in range(size)])

    def set_values():
        for i in range(iterations // 5):
            i = random.randrange(M.nrows())
            j = random.randrange(M.ncols())
            if random.uniform(0, 1) > 0.5:
                # Bigger than 2**62:
                M[i,j] = 10**128
            else:
                # Smaller than 2**62:
                M[i,j] = 0

    def get_dets():
        for _ in range(iterations):
            M.det()

    threads = [Thread(target=set_values) for _ in range(nthreads-1)]
    threads.append(Thread(target=get_dets))

    for t in threads:
        t.start()
    for t in threads:
        t.join()


def test_all_tests():
    test_funcs = {f for name, f in globals().items() if name.startswith("test_")}
    untested = test_funcs - set(all_tests)
    assert not untested, f"Untested functions: {untested}"


all_tests = [
    test_pyflint,
    test_showgood,

    test_fmpz,
    test_fmpz_factor,
    test_fmpz_functions,
    test_fmpz_poly,
    test_fmpz_poly_factor,
    test_fmpz_poly_functions,
    test_fmpz_mat,
    test_fmpz_series,

    test_fmpq,
    test_fmpq_poly,
    test_fmpq_mat,
    test_fmpq_series,

    test_nmod,
    test_nmod_poly,
    test_nmod_mat,
    test_nmod_series,

    test_fmpz_mod,
    test_fmpz_mod_dlog,
    test_fmpz_mod_poly,
    test_fmpz_mod_mat,

    test_division_scalar,
    test_division_poly,
    test_division_matrix,

    test_properties_poly_mpoly,
    test_factor_poly_mpoly,
    test_division_poly_mpoly,

    test_polys,
    test_mpolys_constructor,
    test_mpolys_properties,

    test_poly_resultants,

    test_fmpz_mpoly_vec,

    test_matrices_eq,
    test_matrices_constructor,
    test_matrices_strrepr,
    test_matrices_getitem,
    test_matrices_setitem,
    test_matrices_bool,
    test_matrices_transpose,
    test_matrices_pos_neg,
    test_matrices_add,
    test_matrices_sub,
    test_matrices_mul,
    test_matrices_pow,
    test_matrices_div,
    test_matrices_properties,
    test_matrices_inv,
    test_matrices_det,
    test_matrices_charpoly,
    test_matrices_minpoly,
    test_matrices_rank,
    test_matrices_rref,
    test_matrices_solve,
    test_matrices_fflu,

    test_fq_default,
    test_fq_default_poly,

    test_arb,
    test_acb,

    test_pickling,

    test_python_threads,

    test_all_tests,
]