"""

Tests specific to the bezier module.

"""

import pytest

import numpy as np

from numpy.testing import assert_allclose

from matplotlib.bezier import (

_real_roots_in_01, inside_circle, split_bezier_intersecting_with_closedpath

)

@pytest.mark.parametrize("roots, expected_in_01", [

([0.5], [0.5]),

([0.25, 0.75], [0.25, 0.75]),

([0.2, 0.5, 0.8], [0.2, 0.5, 0.8]),

([0.1, 0.2, 0.3, 0.4], [0.1, 0.2, 0.3, 0.4]),

([0.0, 0.5], [0.0, 0.5]),

([0.5, 1.0], [0.5, 1.0]),

([2.0], []), # outside [0, 1]

([0.5, 2.0], [0.5]), # one in, one out

([-1j, 1j], []), # complex roots

([0.5, -1j, 1j], [0.5]), # mix of real and complex

([0.3, 0.3], [0.3, 0.3]), # repeated root

])

def test_real_roots_in_01(roots, expected_in_01):

roots = np.array(roots)

coeffs = np.poly(roots)[::-1] # np.poly gives descending, we need ascending

result = _real_roots_in_01(coeffs.real)

assert_allclose(result, expected_in_01, atol=1e-10)

@pytest.mark.parametrize("coeffs", [[5], [0, 0, 0]])

def test_real_roots_in_01_no_roots(coeffs):

assert len(_real_roots_in_01(coeffs)) == 0

def test_split_bezier_with_large_values():

# These numbers come from gh-27753

arrow_path = [(96950809781500.0, 804.7503795623779),

(96950809781500.0, 859.6242585800646),

(96950809781500.0, 914.4981375977513)]

in_f = inside_circle(96950809781500.0, 804.7503795623779, 0.06)

split_bezier_intersecting_with_closedpath(arrow_path, in_f)

# All we are testing is that this completes

# The failure case is an infinite loop resulting from floating point precision

# pytest will timeout if that occurs