a ;Za`@sddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZmZddgdd gd d gd d ggZeeZgd Zgd Zejd d dgejdddgejdddgejdddgddZddZd5ddZdd Zd!d"Zd#Zejd$d%d&ed'fgd(d)Zejd*ded gd+d,Zejd-ed.fd/gd0d1Zejd2ded gd3d4Z dS)6N)sparse)stats)l1_min_c) LinearSVC)LogisticRegression set_seed_wrapbounded_rand_int_wrap)rr r r )r rrloss squared_hingelogX_labelrdenseY_label two-classes multi-classintercept_label no-intercept fit-interceptc CsXttd}ttd}ddidddd}||}||}||} t|||fi| dS) N)rr)rr fit_interceptFT )rintercept_scaling)rr)sparse_Xdense_XY1Y2check_l1_min_c) r rrrZXsZYsZ interceptsXYZintercept_paramsr"r;cCs6ttt|Wdn1s(0YdS)z=Test that `set_seed_wrap` is defined for unsigned 32bits intsN)r)r* OverflowErrorr)r;r"r"r#test_newrand_set_seed_overflow`s r@z range_, n_ptsi')r:c sd}g}tjdd}t|D]4}fddt|D}t||j}||jqtjddd}t||j} | jdksJd| jd tj|d d } | dksJd | d dS)z;Test that `bounded_rand_int` follows a uniform distributionr:r)ZlocZscalecsg|] }tqSr")r ).0_range_r"r# pz1test_newrand_bounded_rand_int..r g?zNull hypothesis rejected: generated random numbers are not uniform. Details: the (meta) p-value of the test of uniform distribution of p-values is z which is not > 0.05r)qzlNull hypothesis rejected: generated random numbers are not uniform. Details: lower 10th quantile p-value of z not > 0.05.N) rZuniformrangeZkstestZcdfappendZpvaluer1Z percentile) rEZn_ptsZn_iterZks_pvalsZ uniform_distrCZsampleresZuniform_p_vals_distZ res_pvalsZmin_10pct_pvalr"rDr#test_newrand_bounded_rand_intgs(   rLrEcCs6ttt|Wdn1s(0YdS)zETest that `bounded_rand_int_wrap` is defined for unsigned 32bits intsN)r)r*r?r rDr"r"r#$test_newrand_bounded_rand_int_limitss rM)TN)!Znumpyr1ZscipyrZsprr)Zsklearn.svm._boundsrZ sklearn.svmrZsklearn.linear_modelrZsklearn.svm._newrandrr rZ csr_matrixrrrZmarkZ parametrizer$r-rr5r6Z_MAX_UNSIGNED_INTr>r@rLrMr"r"r"r#s:          $