U e{j@sDddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZmZmZedddddgZiZedddddgee <edd dd dgee dd e dd <eddd ddgee<edddddgee<eddd ddgeed<edddddgee<edddddgee <edddddgee<edddddgee<edddd d!geed"d#d$<eeee<eeeeddd%<edd&dd'dgeeed<Gd(d)d)Zd*d+Zejd,d-d%dgd.d/Zejd,d-d%dgd0d1Zejd,d-d%dd2gd3d4Zd5d6ZdS)7N) assert_equal) AsinhStretchContrastBiasStretch HistEqStretchInvertedHistEqStretchInvertedLogStretchInvertedPowerDistStretch LinearStretch LogStretchPowerDistStretch PowerStretch SinhStretch SqrtStretchSquaredStretch????g?g?)Z intercept)Zslope'e?g+Pz?g?g?gUJr?gc?g>?g×?gt\J?gM1AG?gzlˀ?guYLl>?g?:?g3}ƅ?gADj4?gWL?g333333ӿg?gffffff?g333333?g333333?g@g?)ZcontrastZbiasg?g}w(?c@seZdZejdeddZejdddgejdeddZ ejded d Z ejded d Z ejded dZ ejdeddZ ejdeddZddZddZdS) TestStretchstretchcCs"tjj|tddt|dddS)NFclipư>Zatol)nptestingassert_allcloseDATARESULTSselfrr$g/var/www/html/bwave/python/venv/lib/python3.8/site-packages/astropy/visualization/tests/test_stretch.py test_no_clip.s  zTestStretch.test_no_clipndimcCsDtjd|}tjj|t|ddtt|dddddS)N)Trrr*rr) r shaperrrZreshapeZravelrr!)r#rr'Z new_shaper$r$r%test_clip_ndimensional4s z"TestStretch.test_clip_ndimensionalcCs,tjj|tddtt|dddddS)NTrrr*rr)rrrr rr!r"r$r$r% test_clip?s  zTestStretch.test_clipcCsJt}ttj}|||ddtjj|t|ddtj|tdS)NFoutrrr)r copyrzerosr+rrr!)r#rZdata_inresultr$r$r% test_inplaceEs  zTestStretch.test_inplacecCs$tj|j|tddddtdS)NFrrrrinverser r"r$r$r%test_round_tripMszTestStretch.test_round_tripcCs<ttj}|t|dd|j||ddtj|tdS)NFr.)rr1r r+r5rr)r#rr2r$r$r%test_inplace_roundtripSs z"TestStretch.test_inplace_roundtripcCs"tjj|jt|tdddS)Nrrr4r"r$r$r%test_double_inverseZs  zTestStretch.test_double_inversecCs*tj}td}tj|t|tdS)Nr()rr5r rrrr )r# stretch_1 stretch_2r$r$r% test_inverted`szTestStretch.test_invertedcCsNtt}td}td}tj|t|ttj|t|tdS)Nrg@)rr rrrr r5)r#r9r:Z stretch_3r$r$r% test_chaininges  zTestStretch.test_chainingN)__name__ __module__ __qualname__pytestmark parametrizer!keysr&r,r-r3r6r7r8r;r<r$r$r$r%r-s"       rcCsdt}|dddddg}tj|dddddg|dddddgdd}tj|tjdddd gdS) Ngrrr?rFrg#?)rrrrnan)rvaluesr$r$r%test_clip_invalidos rGagc CsPd}tjt|dt|dW5QRXtjt|dt|dW5QRXdS)Nz&a must be >= 0, but cannot be set to 1matchrH)r@raises ValueErrorr rrHrJr$r$r%test_invalid_powerdist_ays rOc Cstd}tjt|dt|dW5QRXtjt|dt|dW5QRXtjt|dt|dW5QRXdS)Nz a must be > 0rIrK)r@rLrMr r rrNr$r$r%test_invalid_power_log_asrPrDc CsPd}tjt|dt|dW5QRXtjt|dt|dW5QRXdS)Nza must be > 0 and <= 1rIrK)r@rLrMrr rNr$r$r%test_invalid_sinh_as rQc Cs^ttj dddddtjg}tdddddddg}tt|||tt|||dS)Nrrrrr)rarrayinfrrr)datar2r$r$r%test_histeqstretch_invalidsrU)numpyrr@Z numpy.testingrZastropy.visualization.stretchrrrrrrr r r r r rrrRr r!rrGrArBrOrPrQrUr$r$r$r%s> <    B