U yc@sddlZddlZddZdeeeedddZdeeeeddd Zdeeeeed d d ZeZdddZdddZ dddZ d ddZ d!ddZ d"ddZ dS)#NcCs&||}t|d||}||fS)z4finds mu and sigma corresponding to a Binomial(n, p))mathsqrt)npmusigmar B/home/cheung/ML/DS-from-scratch/07-hypo-testing/lib/HypoTesting.py normal_approximation_to_binomial sr r)xrrreturncCs(t||d d|dt|S)N)rexpZ SQRT_TWO_PIr rrr r r normal_pdfsrcCs$dt||td|dS)Nrr)rerfrrr r r normal_cdfsrh㈵>)rrr tolerancer cCsd|dks|dkr$||t||dSd}d}|||kr`||d}t|}||krZ|}q,|}q,|S)z,Find approximate inverse using binary searchrr)rg$g$@r)inverse_normal_cdfr)rrrrZlow_zZhi_zZmid_zZmid_pr r r r$s  rcCsdt|||SNrr)lorrr r r normal_probability_aboveDsrcCst|||t|||S)Nrrhirrr r r normal_probability_betweenJsrcCsdt||||Sr)rrr r r normal_probability_outsidePsrcCs t|||S)z9 returns the z for which P(Z <= z) = probability r probabilityrrr r r normal_upper_boundXsr"cCstd|||S)z9 returns the z for which P(Z >= z) = probability rrr r r r normal_lower_boundasr#cCs,d|d}t|||}t|||}||fS)zb returns the symmetric (about the mean) bounds that contain the specified probability rr)r#r")r!rrZtail_probabilityZ upper_boundZ lower_boundr r r normal_two_sided_boundsls   r$)rr)rr)rrr)rr)rr)rr)rr)rr)rr)rrandomr floatrrrZnormal_probability_belowrrrr"r#r$r r r r s&