o
i @ 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 ?r5 r5 r
)r6 r r
r. r r. r( ) ?r9 r9 r7 r7 r7 )@r: )r r9 r )r r6 r r ) r
r r
r
r
r
r
r5 r5 r5 r
r6 ) r6 r r
r
r
r
r
r. r7 r r8 r. r r r r
)r r r r r) r
r* diff)r r r r+ Zw2y2Zres2r 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 r6 r4 r r
r r7 r )
r 8T笪@r> r> r> r> r> 8T笪@r? r? )r r
r r. d r r
F)numZendpointr4 r. )dg g_ӽ-?gO*e ?g%
2g?g;ك?g6Y:tW?g>l&?g?g:ǰ?g{?g!:Fv?gEr/?gk;?$'
"?rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB rB 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@rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC rC r/ ) r r r) r r* r r0 sinallr; sum)r r r+ Zx_Rr2 Zres_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 )Nr6 r Fr( ) r r1 floatr' flagsZ writeabler rE 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 )Nr. r r6 r( )
r r1 rH r' rI c_contiguous f_contiguousr rE rJ r) r
r* rK r r r test_non_contiguous_arrays s z1TestIsotonicRegression.test_non_contiguous_arrays)__name__
__module__ __qualname__r markZ parametrizer r$ r r Zfloat32Zint64Zint32r, r3 r= rG rL rO r r r r r sF
8 r ) numpyr Z
numpy.testingr r r Zscipy.optimize._pava_pybindr Zscipy.optimizer r r r r r s