o
h @ sJ d dl Zd dlmZmZ d dlZd dlmZ d dlm Z G dd dZ
dS ) N)assert_allclose assert_equal)pava)isotonic_regressionc @ s e Zd Zejdddggddfddgddggdfddgdgdfdddgdfddgddfddgddgdfgd d
Zd d Zejd
ej ej
ej ej gejdej ej
ej ej gejdddgdd Z
ejdddgdd Zdd Zdd Zd d Zd d ZdS ) TestIsotonicRegression)ywmsgr Nz7array has incorrect number of dimensions: 2; expected 1 z s
z$TestIsotonicRegression.test_linspacec C s t g d}t g d}t||d}t|jg d t|jg d t|jg d t g d}t g d}t||d}tt |jd d
d t|jd d |j t|j|j t|jd d d |jd d d S )
N)r
r ?r8 r8 r
)r9 r r
r1 r r1 r+ ) ?r< r< r: r: r: )@r= )r r< r )r r9 r r ) r
r r
r
r
r
r
r8 r8 r8 r
r9 ) r9 r r
r
r
r
r
r1 r: r r; r1 r r r r
)r r! r r r, r
r- diff)r r r r. w2y2res2r r r test_weightsF s &z#TestIsotonicRegression.test_weightsc C s g d}t |}g d}t|j| t|jg d d}tjdd|dd}d | td
| }t |}g d }t|j| tt |jdksKJ tt
|jt
| t | dd }t|j |j t|j|j d S )
N)
r r r r9 r7 r r
r r: r )
r 8T笪@rC rC rC rC rC 8T笪@rD rD )r r
r r1 d r r
F)numendpointr7 r1 )dg g_ӽ-?gO*e ?g%
2g?g;ك?g6Y:tW?g>l&?g?g:ǰ?g{?g!:Fv?gEr/?gk;?$'
"?rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH rH gV ?gQk?v?g0n?g l@goT@gZO"@gN%U @glJ
@gt NԐ @go @gDN> @gVi$'@glO @gc@g] @gǾ @g{ur @gfue@g@gkY@gHZ@DFjv@rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI r2 ) r r r, r r- r r3 sinallr> sum)r r r. x_Rr5 res_invr r r test_against_R_monotoneX s z.TestIsotonicRegression.test_against_R_monotonec C s~ t jdtd}t jdtd}d|j_d|j_t||d}t t |j s'J t t |j
s2J t t |j s=J d S )Nr9 r Fr+ ) r r4 floatr* flags writeabler rK isfiniter, r
r- r r, r r. r r r
test_readonly s z$TestIsotonicRegression.test_readonlyc C s t jdtdd d d }t jdtdd d d }|jjr J |jjr$J |jjr*J |jjr0J t||d}t t |j
sAJ t t |j sLJ t t |j sWJ d S )Nr1 r r9 r+ )
r r4 rP r* rQ c_contiguous f_contiguousr rK rS r, r
r- rT r r r test_non_contiguous_arrays s z1TestIsotonicRegression.test_non_contiguous_arrays)__name__
__module__ __qualname__r mark parametrizer r' r r" float32int64int32r/ r6 rB rO rU rX r r r r r sF
8 r ) numpyr
numpy.testingr r r scipy.optimize._pava_pybindr scipy.optimizer r r r r r s