Q(h P d dl Zd dlmZmZ d dlZd dlmZ d dlm Z G d d Z
y) N)assert_allclose assert_equal)pava)isotonic_regressionc e Zd Zej j dddggddfddgddggdfddgdgdfdddgdfddgddfddgddgdfg d Zd
Zej j d ej ej ej ej g ej j d ej ej ej ej g ej j d
ddg d Z
ej j dddg d Zd Zd Zd Zd Zy)TestIsotonicRegression)ywmsgr Nz7array has incorrect number of dimensions: 2; expected 1 z s_ $.BKK1a BKK1a4H!!
;q!
BIIa!e$45r+ c t j g d }t j g d }t || }t |j g d t |j
g d t |j g d t j g d }t j g d }t || }t t j |j d d
d t |j d d |j t |j
|j
t |j d d d z
|j d d y )
N)r r
?r? r? r )r@ r
r r7 r r7 r0 ) ?rC rC rA rA rA )@rD )r rC r )r r@ r! r ) r r
r r r r r r? r? r? r r@ ) r@ r
r r r r r r7 rA r rB r7 r r" r r )r# r$ r r r1 r r2 diff)r r
r r3 w2y2res2s r test_weightsz#TestIsotonicRegression.test_weightsF s HH3 4 HH0 1!!Q/ FG [1
L1 XX@
A
XX=
>"2r2q ,a0qr
CEE* ckk2 AB!+SZZ^ r
r r" rA r )
r 8T笪@rK rK rK rK rK 8T笪@rL rL )r r r" r7 d r r F)numendpointr> r7 )dg g_ӽ-?gO*e ?g%
2g?g;ك?g6Y:tW?g>l&?g?g:ǰ?g{?g!:Fv?gEr/?gk;?$'
"?rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP rP 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@rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ rQ r8 ) r r r1 r r2 r# r9 sinallrE sum)r r r3 x_Rr; res_invs r test_against_R_monotonez.TestIsotonicRegression.test_against_R_monotoneX s *!!$
s#SZZ / KK1!e 4
EBFF26N "!!$
& s# vvbggceen)*** suu
rvvay1 &qbU;
CEE*W^^SZZ0r+ c t j dt }t j dt }d|j _ d|j _ t
|| }t j t j |j sJ t j t j |j sJ t j t j |j sJ y )Nr@ r Fr0 ) r# r: floatr. flags writeabler rS isfiniter1 r r2 r r1 r
r3 s r
test_readonlyz$TestIsotonicRegression.test_readonly s IIau % GGAU # ! !!!Q/vvbkk#%%()))vvbkk#++.///vvbkk#**-...r+ c t j dt d d d }t j dt d d d }|j j
rJ |j j rJ |j j
rJ |j j rJ t || }t j t j |j sJ t j t j |j sJ t j t j |j sJ y )Nr7 r r@ r0 )
r# r: rY r. rZ c_contiguous f_contiguousr rS r\ r1 r r2 r] s r test_non_contiguous_arraysz1TestIsotonicRegression.test_non_contiguous_arrays s IIb &ss + GGBe $SqS )77''''77''''77''''77''''!!Q/vvbkk#%%()))vvbkk#++.///vvbkk#**-...r+ )__name__
__module__ __qualname__r mark parametrizer r* r# r% float32int64int32r4 r< rI rW r^ rb r+ r r r sx [[ !fXt
F
HVq!fX
K
MVaS
K
M
A
K
MVQ
K
MVaV
2
4
"0#"0 : [[ YRZZ288(TU [[ YRZZ288(TU [[ S4.1
K 2 V V
K [[ \D%=96 :6=$61p
/ /r+ r ) numpyr#
numpy.testingr r r scipy.optimize._pava_pybindr scipy.optimizer r rk r+ r rp s 7
, .^/ ^/r+