^Mh|'jddlZddlZddlmZmZddlmZm Z GddZ GddZ dS)N) assert_equalassert_allclose) _iv_ratio _iv_ratio_cceZdZejdgddZejddejdfejddfgdZ ejddej ej ejgejd ej e j ej e j ej ej ejgd Zejdd dej e jejgd Zejd dej e j fdej e jfdej e jdzfdej e jdfej e jejej e jfgdZejd ddejej e jej e jfgdZejd ej e jej e jfej e jdz ej e jfej e jej e jdz fgdZdS) TestIvRatiov,x,r))?UUUUUU?g.a0R#?)r UUUUUU?g<)?)r r gVS?)r UUUUUU?gP]k(?)r 笪?gjD?)6Z5Z?g&R͒U?)r窪?gZ?)r竪?gZr!?)r?g4e~u?)r}| @gG)ȿ?)Q@}P?g1a?)rj6i?gִN`?)r:m@g9Ƭ7?)r5T@g4+?)r翎H%@gJ]?)EdL@9L;w3@g'~V?)r^s!iFE@g/X?)rSR@g_8?)rPT`@g)X?)r>=s@g\h*?cHtt|||dddS)aThe reference values are computed using mpmath as follows. from mpmath import mp mp.dps = 100 def iv_ratio_mp(v, x): return mp.besseli(v, x) / mp.besseli(v - 1, x) def _sample(n, *, v): '''Return n positive real numbers x such that iv_ratio(v, x) are roughly evenly spaced over (0, 1). The formula is taken from [1]. [1] Banerjee A., Dhillon, I. S., Ghosh, J., Sra, S. (2005). "Clustering on the Unit Hypersphere using von Mises-Fisher Distributions." Journal of Machine Learning Research, 6(46):1345-1382. ''' r = np.arange(1, n+1) / (n+1) return r * (2*v-r*r) / (1-r*r) for v in (0.5, 1, 2.34, 56.789): xs = _sample(5, v=v) for x in xs: print(f"({v}, {x}, {float(iv_ratio_mp(v,x))}),") 缉ؗҼ>>>>>rrcBtt|||dSziIf exactly one of v or x is inf and the other is within domain, should return 0 or 1 accordingly.Nrr&r's r,test_infzTestIvRatio.test_inf@s" Xa^^Q'''''r.r)\(\?r*cVtt||tjdSzeIf at least one argument is out of domain, or if v = x = inf, the function should return nan.N)rr&npnanr(r)r*s r,test_nanzTestIvRatio.test_nanIs$ Xa^^RV,,,,,r.r c~tt|ddtt|dddS)z?If x is +/-0.0, return x to ensure iv_ratio is an odd function.Nr1r(r)s r, test_zero_xzTestIvRatio.test_zero_xRs> Xa%%s+++Xa&&-----r.v,x@xD{cNtt||d|z|z dS)a9If x is much less than v, the bounds x x --------------------------- <= R <= ----------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-1+sqrt(x**2+(v+1)**2) collapses to R ~= x/2v. Test against this asymptotic expression. r Nr1r8s r, test_tiny_xzTestIvRatio.test_tiny_xXs*" Xa^^c!eQY/////r.rg7yACrBg\)c=HcBtt||ddS)aAIf x is much greater than v, the bounds x x --------------------------- <= R <= --------------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-0.5+sqrt(x**2+(v-0.5)**2) collapses to R ~= 1. Test against this asymptotic expression. ?Nr1r8s r, test_huge_xzTestIvRatio.test_huge_xks" Xa^^S)))))r.c||z }|dtjd|zz }tt|||dddS)aIf both x and v are very large, the bounds x x --------------------------- <= R <= ----------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-1+sqrt(x**2+(v+1)**2) collapses to R ~= x/(v+sqrt(x**2+v**2). Test against this asymptotic expression, and in particular that no numerical overflow occurs during intermediate calculations. rr"rr#N)r6hypotrr&r(r)r*texpecteds r, test_huge_v_xzTestIvRatio.test_huge_v_x{sK EBHQNN*+Au1EEEEEEr.N__name__ __module__ __qualname__pytestmark parametrizer-r6infr2r7finfofloatsmallest_normalsmallest_subnormalr9maxr>sqrtrErJrQr.r,rr sY [W''',??-,?8 [W BFA A'(( (  [S4"&"&"&"ABB [SHBHUOO$C#C$,BHUOO$F#F$&F7BFBF#<==--==CB-  [S3828E??+>"GHH..IH.  [U HBHUOO +, HBHUOO ./ HBHUOO .q 01 % a % gbghbhuoo&9::; % 0 0 0 [U %$ % %xrx':;% * *  * [U % hbhuoo12 % q ("(5//"56 % hbhuoo1A56% F F  F F Fr.rceZdZejdgddZejddejdfejddfgdZ ejddej ej ejgejd ej e j ej e j ej ej ejgd Zejdd dej e jejgd Zejd dej e j fdej e jfdej e jdzfdej e jdfej e jejej e jfgdZejd ddejej e jej e jfgdZejd ej e jej e jfej e jdz ej e jfej e jej e jdz fgdZdS) TestIvRatioCr ))r r g{s+?)r r ga*?)r r gTV6?)r r g`D)?)r rg,wU?)rrgvL[?)rrg>7R?)rrgL?)rrg5?)rrgZ?)rrg3zʈ?)rrgؤO??)rrgsF?)rrg1?)rrgL9Ԋ?)rrgCv`?)rrg -S?)rrg@ɣ?)rrgO?)rr g^VO?cHtt|||dddS)z8The reference values are one minus those of TestIvRatio.V瞯zTestIvRatioC.test_zero_xs> Z3''---Z4((#.....r.r?r@rAcTtt||dd|z|z z dS)a=If x is much less than v, the bounds x x --------------------------- <= R <= ----------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-1+sqrt(x**2+(v+1)**2) collapses to 1-R ~= 1-x/2v. Test against this asymptotic expression. rIr Nrhr8s r,rEzTestIvRatioC.test_tiny_xs0" Z1%%sCE19}55555r.rFrGcTtt|||dz |z dddS)aKIf x is much greater than v, the bounds x x --------------------------- <= R <= --------------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-0.5+sqrt(x**2+(v-0.5)**2) collapses to 1-R ~= (v-0.5)/x. Test against this asymptotic expression. r rdrr#Nrer8s r,rJzTestIvRatioC.test_huge_xs3  1a((1S5!)%aHHHHHHr.rKc||z }d|dtjd|zz z }tt|||dddS)aIf both x and v are very large, the bounds x x --------------------------- <= R <= ----------------------- v-0.5+sqrt(x**2+(v+0.5)**2) v-1+sqrt(x**2+(v+1)**2) collapses to 1 - R ~= 1 - x/(v+sqrt(x**2+v**2). Test against this asymptotic expression, and in particular that no numerical overflow occurs during intermediate calculations. rr"rr#N)r6rMrrfrNs r,rQzTestIvRatioC.test_huge_v_xsR EqAA.// 1a(((QGGGGGGr.NrRr`r.r,rbrbs_ [W''',AA-,A [W BFA A'** *  [S4"&"&"&"ABB [SHBHUOO$C#C$,BHUOO$F#F$&F7BFBF#<==//==CB/  [S3828E??+>"GHH//IH/  [U HBHUOO +, HBHUOO ./ HBHUOO .q 01 % a % gbghbhuoo&9::; % 6 6 6 [U %$ % %xrx':;% I I  I [U % hbhuoo12 % q ("(5//"56 % hbhuoo1A56% H H  H H Hr.rb) rVnumpyr6 numpy.testingrrscipy.special._ufuncsrr&rrfrrbr`r.r,rqs 77777777 AFAFAFAFAFAFAFAFHiHiHiHiHiHiHiHiHiHiHr.