o
4Bi^ @ 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 Zejdddggddfddgddggdfddgdgdfdddgddgd fgd
d Zd d
Zejdej 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 )r r= r )r r: r r ) r
r r
r
r
r
r
r9 r9 r9 r
r: ) r: r r
r
r
r
r
r2 r; r r< r2 r r r r
)r! r" r r r- r r. diff)r r r r/ w2y2res2r r r test_weightsD 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 r: r8 r r
r r; r )
r 8T笪@rD rD rD rD rD 8T笪@rE rE )r r
r r2 d r r
F)numendpointr8 r2 )dg g_ӽ-?gO*e ?g%
2g?g;ك?g6Y:tW?g>l&?g?g:ǰ?g{?g!:Fv?gEr/?gk;?$'
"?rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI rI 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@rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ rJ r3 ) r r r- r r. r! r4 sinallr? sum)r r r/ x_Rr6 res_invr r r test_against_R_monotoneV 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 )Nr: r Fr, ) r! r5 floatr+ flags writeabler rL 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 )Nr2 r r: r, )
r! r5 rQ r+ rR c_contiguous f_contiguousr rL rT r- r r. rU r r r test_non_contiguous_arrays s z1TestIsotonicRegression.test_non_contiguous_arrays)__name__
__module__ __qualname__r mark parametrizer r( r! r# float32int64int32r0 r7 rC rP rV rY r r r r r s<
8 r ) numpyr!
numpy.testingr r r scipy.optimize._pava_pybindr scipy.optimizer r r r r r s