o h @sddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZejd ejejgd d ZddZddZddZddZ ddZ!ddZ"ddZ#ddZ$ejd ej%ejej&gddZ'ejd ej%ejej&gd d!Z(d"d#Z)ejd$d d%gejd ej*ej+ej&gejd&gd'd(d)Z,ej-.de /Z0e 1e0Z2e 3e2d*d+Z2d,d-Z4d.d/Z5ejd ej%ejej&gd0d1Z6d2d3Z7d4d5Z8ejd6d%dd7gfd dd8gfd9d:d;gfgdd?gfd d@d7gfdAdBdCgfgdDdEZ:dFdGZ;dHZdKZ?dLdMZ@dNdOZAejd ej%ejej&gdPdQZBdRdSZCejd ejejDej%ejej&gdTdUZEdVdWZFdXdYZGdZd[ZHd\d]ZId^d_ZJd`daZKdbdcZLejdddedfgdgdhZMejdidjej*fdkejNfdlejNfdmejNfdneOfgdodpZPdqdrZQdsdtZRdudvZSejd ej%ejej&gdwdxZTdydzZUd{d|ZVd}d~ZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ejd ej%ejej&gddZ`ddZaddZbddZcddZdejd ej%ejej&gddZeddZfddZgejd ej%ejej&gddZhddZiddZjddZkddZlddZmddZnejde joe jpe jqgddZrddZsddZtdS)N)assert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal assert_equal)data)exposure)util)rgb2gray)intensity_range) dtype_range)expected_warnings)_supported_float_typedtypecCsXtjddg|d}tjtddtj|dd\}}WddS1s%wYdS)Ndrz+Incorrect value for `source_range` argumentmatchfoobar source_range)nparraypytestraises ValueErrorr histogramrim frequencies bin_centersr"_/var/www/html/scripts/venv/lib/python3.10/site-packages/skimage/exposure/tests/test_exposure.pytest_wrong_source_ranges  "r$cCsjtjddgtjd}t|\}}t|tdd|ddks"J|ddks*Jt|ddddS)Nrrrerrrint8rrrarangerr r!r"r"r#test_negative_overflow r+cCsjtjddgtjd}t|\}}t|tdd|ddks"J|ddks*Jt|ddddS)Nirrrr&r'r*r"r"r#test_all_negative_image)r,r-cCsTtjddgtjd}t|\}}tt|t|t|ddt|dddS)N rrrr)rrr(rrrlenr*r"r"r#test_int_range_image2s r0cCsxtjddgddggtjd}tj|dd\}}t|jdD] }tt||t|qt|ddt|dddS) Nr.rfrr) channel_axisr) rrr(rrrangeshaperr/)rr r!chr"r"r#!test_multichannel_int_range_image:s r7cCsrtjddgtjd}tj|dd\}}t|tddt|ddt|ddt|d dt|jd dS) Nr.rrrrrr&r%r8) rruint8rrrr)rr5r*r"r"r#test_peak_uint_range_dtypeCr;cCsrtjddgtjd}tj|dd\}}t|tddt|dd t|d d t|d d t|jd dS)Nr.rrrrr&rr9) rrr(rrrr)rr5r*r"r"r#test_peak_int_range_dtypeMr<rBcCsHtjdddtjd}tj|dd\}}t|tddt|jddS)Nrr8rrrr9) rlinspacer:rrrr)rr5r*r"r"r#test_flat_uint_range_dtypeWrEcCsHtjdddtjd}tj|dd\}}t|tddt|jddS)Nr=r>r8rrrr9) rrDr(rrrr)rr5r*r"r"r#test_flat_int_range_dtype^rFrGcCsLtjddg|d}tj|dd\}}|j|ksJt|tddddS)Nr.rrZnbins?)rrrrrrr)rr"r"r#"test_peak_float_out_of_range_imageesrLcCsntjddg|d}d}tj||dd\}}|j|ksJtt|ddtt|ddtt |ddS) Nr.rrr)rJrg?) rrrrrrminmaxrr/)rrrJr r!r"r"r#"test_peak_float_out_of_range_dtypens rQcCsvtjgdtjd}tj|ddd\}}td}d|d<d |d <t||tj|dd d\}}|d }t||dS) N)rrCrCrrF)r normalizer8r&rrTg@)rrr:rrzerosr)rr r!expectedr"r"r#test_normalizezs   rVrimager3)rr&rc Csd}|d|d}t|\}}tjtj|||dtj|||df|d}tj|||d\}} t|tjrAt| t ||d|dd|ksKJ|dddksUJ|dddks_J|dd|ksiJdS) z-Check that all channels use the same binning.)r1r1rr&r)axis)rr3rN) r rstackfullrr issubdtypeintegerrr)) rrr3r5 channel_sizeiminimaxrr r!r"r"r#(test_multichannel_hist_common_bins_uint8s$  r`g@rcCs&tt}tjtdd}t||dS)z)Check integer bins used for uint8 images.rMrIN)r equalize_hist test_img_intr)img_eq0img_eq1r"r"r#test_equalize_uint8_approxs recCs.tt}t|}t|\}}t|dS)N)r img_as_ubytetest_imgrracumulative_distributioncheck_cdf_slope)imgimg_eqcdf bin_edgesr"r"r#test_equalize_ubytes   rncCs\ttj|dd}t|}|jt|ksJt|\}}t ||jt|ks,JdS)NFcopy) r img_as_floatrgastyperrarrrhri)rrjrkrlrmr"r"r#test_equalize_floats  rscCsltt}ttj}d|ddddf<tj||d}t|}t|\}}t |||k r4JdS)Nr&ri)mask) r rqrgrrTr5rrarhriall)rjrt img_mask_eqrkrlrmr"r"r#test_equalize_maskeds   rwcCsDtddt|}t||d\}}d|krdks JJdS)z=Slope of cdf which should equal 1 for an equalized histogram.rr&rN皙?N)rrDr/polyfit)rlnorm_intensityslope interceptr"r"r#ris riztest_input,expectedr&rC)r.r.r}cC.tjddgtjd}t||d}t||dS)Nrr&r range_values)rrr:r r test_inputrUrWoutr"r"r#test_intensity_range_uint8 r皙?皙?r)333333?皙?rrcCr~)Nrrrrrrfloat64r rrr"r"r#test_intensity_range_floatrrcCs0tjddgtjd}t|ddd}t|ddS)NrrrrT)r clip_negative)rr&rrWrr"r"r#"test_intensity_range_clipped_floatsriii?icCs@tjgdtjd}t|}|jtjksJt|gddS)N3r2rrrC)rrr:rrescale_intensityrrrr"r"r#test_rescale_stretch s rcCs*tgd}t|}t|gddS)NgI@gY@g c@)rrKr&rrrrrrr"r"r#test_rescale_shrinks rcCsFtjgd|d}tj|dd}t|gddd|j|jks!JdS)NrrrrCin_range)rrg333333?)decimal)rrrrrrrrWrr"r"r#test_rescale_in_rangesrcCs.tgd}tj|dd}t|gddS)Nr)rr2r)rKr&r&rrr"r"r#test_rescale_in_range_clip srcCsFtjgd|d}tj|dd}|jt|jksJt|gddS)a?Check that output range is correct. .. versionchanged:: 0.17 This function used to return dtype matching the input dtype. It now matches the output. .. versionchanged:: 0.19 float16 and float32 inputs now result in float32 output. Formerly they would give float64 outputs. )rr.rrr out_range)rgO@rN)rrrrrrrrr"r"r#test_rescale_out_range&s rcCs<tjdttdgtjd}tj|dd}t|dttgdS)Nrrruint10r)rr uint10_maxuint16rrr uint16_maxrr"r"r#test_rescale_named_in_range9srcC4tjdtgtjd}tj|dd}t|dtgdS)Nrrrr)rrrrrrrrrr"r"r#test_rescale_named_out_range?rcCr)Nrruint12r)rrrrrrr uint12_maxrr"r"r#test_rescale_uint12_limitsErrcCr)Nrruint14r)rrrrrrr uint14_maxrr"r"r#test_rescale_uint14_limitsKrrcCs<tjdtjd}t|}t|sJt||dS)NrSrSr)rrTr:rrisnanrurrr"r"r#test_rescale_all_zerosQs rcCs4tjddgtjd}tj|dd}t|ddgdS)Nrrrr)rrrrrrrr"r"r#test_rescale_constantXrrcCs6td}t|}t|sJt||dS)Nr)ronesrrrrurrr"r"r#test_rescale_same_values^s  rzin_range,out_range)rWr)rrWcCs|tjdtddd}tj|d<d}d}|dkr||g}n|g}t|t|||WddS1s7wYdS) N rrMr)r&r&zWOne or more intensity levels are NaN\. Rescaling will broadcast NaN to the full image\.z/Passing `np.nan` to mean no clipping in np.cliprW)rr)floatreshapenanr rr)rrrWmsgnumpy_warning_1_17_plusexp_warnr"r"r#test_rescale_nan_warninges    "rzout_range, out_dtyper:rrrrcCs4tjgdtjd}tj||d}|j|ksJdS)Nr=rrrr)rrr(rrr)r out_dtyperW output_imager"r"r#test_rescale_output_dtypes rcCsFtjgdtjd}tj|tjd}t|gd|jtjks!JdS)Nrrrrr>rC)rrr(rrr:rrrWrr"r"r#test_rescale_no_overflowsrcCsBtjgdtjd}tj|dd}t|gd|jtksJdS)Nrrrrr)rrr(rrrrrrr"r"r#test_rescale_float_outputsrcCsRtjgdtjd}tttj|dd}WddS1s"wYdS)Nrrflatr)rrr(rrrrr)rW_r"r"r#*test_rescale_raises_on_incorrect_out_ranges "rcCsttj|dd}t|}t|||f}tj |dddd}|j |j ks)J|j t |ks2J|tj kr9dnd}tt||d |tt||d dd S) z!Test a grayscale float image Fro)9r{Gz?r>) kernel_size clip_limitrJrMrSg)\Y@ghs?N)r rqr astronautrrr rdstackrequalize_adapthistr5rrfloat16rpeak_snrnorm_brightness_err)rrjadapted snr_decimalr"r"r#test_adapthist_grayscalesrcCstt}tjdd}tdt|\}}t |dks"JWdn1s,wYtj |dd}| dks@J| dksHJ|j |j ksPJt|}tt||d d tt||d d dS) z#Test an RGB color uint16 image T)recordalwaysrNrr?x&Y[@r&{Gz?rS)r img_as_uintrrwarningscatch_warnings simplefilterrrr/rrOrPr5rrrr)rjwhistr!r full_scaler"r"r#test_adapthist_colors  rcCstt}tj|jd|jdftd}t||f}t |}|j|jks*J|ddddddf}t |}|j|jksDJt t ||ddt t||dddS) zTest an RGBA color image rr&rNrMrrSga+e?)r rqrrrrr5rrrrrrrr)rjalpharrr"r"r#test_adapthist_alphas  rcCstt}t|}d}t|dd|dd|f}t|g|jd|}tj |ddd}tj |ddd}|j|jksAJ|j|jksIJt t |||jdddks^Jd S) z Test for n-dimensional consistency with float images Note: Currently if img.ndim == 3, img.shape[2] > 4 must hold for the image not to be interpreted as a color image by @adapt_rgb rrr1g?)rrrSrN) r rqrrr rrr5rrmeanabs)rjaimg2dimg3d adapted2d adapted3dr"r"r#test_adapthist_grayscale_Nds(  rcCstd}|d7}|tj}t|d}t|t|ks!Jtd}|d7}|tj}t|d}t|t|ksBJdS)z(Test constant image, float and uint rrSrMrN) rrTrrrrrrOrPr)rjrr"r"r#test_adapthist_constants      rcCsttt}|d}d||jdd|jddf<d}tdddD]4}tj||d d }t |d d |f|d d |fd ksDJt ||d d f||d d fd ksYJq%d S) zTest border processing gY@rrrSr&rrGrKrNr) r r rqrrr5r4rrr)rj border_indexrrr"r"r#test_adapthist_borderss"  rcCsbt}t|}tj|dd}tj|dd}t||tj|dd}tj|dd}t||dS)Nrrr&)rmoonr rqrrr)img_uimg_f img_clahe0 img_clahe1r"r"r#test_adapthist_clip_limits  rcCsv|jdkrt|t|}}t|}t|}d|jt||}t |j j \}}dt ||S)zPeak signal to noise ratio of two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- peak_snr : float Peak signal to noise ratio rMrr}) ndimr rpr rqsizersquaresumr rtypelog)img1img2msermax_r"r"r#r,s   rcCsJ|jdkrt|t|}}t||}|t|jjd}|S)zNormalized Absolute Mean Brightness Error between two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- norm_brightness_error : float Normalized absolute mean brightness error rMr&)rr rrrr rr)rrambenber"r"r#rBs rcCsNtjdtd}tjtddt|dWddS1s wYdS)Nrrz Incorrect value of `kernel_size`r)rMrMrM)rrrrrrrrrjr"r"r#$test_adapthist_incorrect_kernel_sizeVs"rcCs.tddg}t|d}|j|jksJdS)"Check that the shape is maintainedr&g?N)rrr adjust_gammar5)rjresultr"r"r#test_adjust_gamma_1x1_shape_s rcCs2tjddtjdd}t|d}t||dS)z4Same image should be returned for gamma equal to onerr8rrr&N)rr)r:rrrr)rWrr"r"r#test_adjust_gamma_onefs rcCsTtjdddj|dd}t|d}|jj}t|t |d|j|jks(JdS)z6White image should be returned for gamma equal to zerorrCrFror&N) rrandomuniformrrrrrrrr )rrWrr"r"r#test_adjust_gamma_zeroms  rcCLtjddtjdd}tjgdtjdd}t|d}t||dS)z\Verifying the output with expected results for gamma correction with gamma equal to halfrr8rr(rr $'*-0257:<>@BDFrIKMNPQSTVWYrH\]^`abrr%r2gijklmopqrstuvwyz{|}~rr>rrWr?r`rfrrjrnrrrurxr{r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrr@r@rArArrrrrrrrrrrrrrrrrrrrrrrrCrKNrr)r:rrrrrrWrUrr"r"r#test_adjust_gamma_less_onews  rcCr)z[Verifying the output with expected results for gamma correction with gamma equal to tworr8rr(rrrrrrrrrrrrr&r&r&r&r&r&r&r&rSrSrSrSrSrSrMrMrMrMrrrrr1r1r1r1rrrrrrr rrr.r. rrrr rrrrrrrrrr}r}rrrrrrrrrr!"#rr%&rr()rr+,r ./rr!1r"r4r#6r$8rrr%;r&=r'?r(Ar)Cr*Er+rr,Jr-Lr.r/Or0r1Rr3Ur4r5Xr6[r7r8r9_r;r<crr2r=hr>r@rArBrCrDrErGrHrIrKxrLrNrOrQrr>rrSrUrVrXrYrZr[r]r^r`rarcrdrfrgrhrirkrlrnrprqrsrtrvrxryr{r}r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrr@rrrrrrrrrrrrrrCrSNrrr"r"r#test_adjust_gamma_greater_ones  rcCsTtdddtjd}ttt|dWddS1s#wYdS)NrrCrrr) rr)r:rrrrrrrWr"r"r#test_adjust_gamma_negatives "rcCs6dtjdtjd}ttj|ddddksJdS)NrCrrr&rx)gammagain)rrr:rurrr r"r"r#test_adjust_gamma_u8_overflows"rcCs@tjddg|d}t|d}|j|jksJ|j|ksJdS)rr&rN)rrr adjust_logr5rrrjrr"r"r#test_adjust_log_1x1_shapes rc Csrtdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|d}t||dS)z|Verifying the output with expected results for logarithmic correction with multiplier constant multiplier equal to unityrrCrr)rr1rrrrrr)rr!r#r%rr*r,r.)rr4rrrrrBrE)rIrLrPrRrUrYr\r`)rdrrkrnrrrvrzr})rrrrrrrr)rrrrrrrr@)rrrrrrrrrr&Nrr)r:rrrrrrr"r"r#test_adjust_logs r c Cttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd}t||dS)zVerifying the output with expected results for inverse logarithmic correction with multiplier constant multiplier equal to unityrrCrr)rrSr1rrrrr})rrrrrrrr )r!rr$r%rrr*H)rrr2r5rHr9r<r2)r?nrFrJrMrQrrV)r?r^rbrfrjrnrsrx)r|rrrrrrr)rrrrArrrrrr&TNr rr"r"r#test_adjust_inv_log rcCsBtjddg|d}t|dd}|j|jksJ|j|ksJdS)rr&rr1N)rrradjust_sigmoidr5rrr"r"r#test_adjust_sigmoid_1x1_shapesrc Cr )zlVerifying the output with expected results for sigmoid correction with cutoff equal to one and gain of 5rrCrr)r&r&r&rSrSrSrSrS)rMrMrMrrrr1r1)r1rrrrrrr.)r.rrrrrrr)rr}rrrrrr)rrrrrrr"r)rrr(r*r rr0r)r6r9rrrArErJrNrr&r1Nrr)r:rrrrrrr"r"r#test_adjust_sigmoid_cutoff_one rrc Csttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd }t||d S)znVerifying the output with expected results for sigmoid correction with cutoff equal to zero and gain of 10rrCrr)rrYrbrjrtr}rr)rrrrrrrAr)rrrrrrrrrrrrrrrr)rrrrrrrr)rrrrrrrrrr.Nrrr"r"r#test_adjust_sigmoid_cutoff_zerorrc Cr )znVerifying the output with expected results for sigmoid correction with cutoff equal to half and gain of 10rrCrr)r&r&rSrSrMrMrr1)r1rrrr.rrr)rrrrrrrr")rr(r r0r6rrArJ)r>r?rcrlrur~rr)rrrrrrrAr)rrrrrrrrrrrKr.Nrrr"r"r#test_adjust_sigmoid_cutoff_half-rrc Csvtdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|ddd}t||dS)zvVerifying the output with expected results for inverse sigmoid correction with cutoff equal to half and gain of 10rrCrr)rrrrrrrr)rrrrrrrr)rrrArrrrr)rrrr|rsrirarX)rQrHr?r:r5r/r+r')r$rrrrrrr)rrrrr.rrr)r1rrrMrMrSrSr&rrKr.TNrrr"r"r##test_adjust_inv_sigmoid_cutoff_half?s rcCstddd}t|sJd|d<t|sJtj|ddr"J|dtj}t|s1Jtj|ddr:J|tjd}t|sIJtj|ddrRJdS) Nrg{Gz?rr&r)upper_percentilerCr8)rrDris_low_contrastrrr:rrr"r"r#test_is_low_contrastQsrcCs:tjdtd}t|sJd|dd<t|rJdS)Nrrr&r1)rrTboolrrrr"r"r#test_is_low_contrast_booleanas r exposure_funccCsTtddddtj}tt ||WddS1s#wYdS)Nrrrr)rr)rrrrrrr)rrWr"r"r#test_negative_inputls  "rcCstjdddddlm}|jtddgddggdd}t|\}}gd }gd }t||s4Jt||ss                                       '#