g RddlmZddlmZmZddlZerddlmZ ddd Z dddZ dS)) annotations) TYPE_CHECKINGAnyN)CoordinateArrayFshapetuple[int, int] want_maskboolreturnVtuple[CoordinateArray, CoordinateArray, CoordinateArray | np.ma.MaskedArray[Any, Any]]c n|\}}tj|tj}tj|tj}tj||\}}|dz }|dz }tjgd}tjddgddgdd gd dgd d gg} tjgd } tj|} t t|D]T} | || tj||z | | d fz dz||z | | dfz dzz | | dzz zz } U|rttj ||z dz dzdz ||z dz dzdz zdk||z dz dzdz ||z dz dzdz zdk} tj | | } ||| fS)aiReturn simple test data consisting of the sum of two gaussians. Args: shape (tuple(int, int)): 2D shape of data to return. want_mask (bool, optional): Whether test data should be masked or not, default ``False``. Return: Tuple of 3 arrays: ``x``, ``y``, ``z`` test data, ``z`` will be masked if ``want_mask=True``. dtype?)rg皙?gffffff?皙?皙?g333333?rg?g?rg?)rrrr皙?rrg{Gz?g?g{Gz?mask) nparangefloat64meshgridasarray zeros_likerangelenexp logical_ormaarray)rr nynxxyxscaleyscaleampmidwidthzirs I/granary/edev/quantum/lib/python3.11/site-packages/contourpy/util/data.pysimpler3 sFB "BJ'''A "BJ'''A ;q!  DAq #XF #XF *000 1 1C *sCj3*sDkC:c{S T TC J000 1 1E aA 3s88__cc SVBFahQT2Q6!F(SAY:NQR9RRSV[\]V^`aVaabb bb&}hnq 3 &!F(S.1)*E E L   EKKK % % a7Nrseedint mask_fractionfloatc|\}}tj|tj}tj|tj}tj||\}}tj|}||}|dkrKt|d}|||k} tj || }|||fS)aReturn random test data in the range 0 to 1. Args: shape (tuple(int, int)): 2D shape of data to return. seed (int, optional): Seed for random number generator, default 2187. mask_fraction (float, optional): Fraction of elements to mask, default 0. Return: Tuple of 3 arrays: ``x``, ``y``, ``z`` test data, ``z`` will be masked if ``mask_fraction`` is greater than zero. r)sizergGz?r) rrrrrandom default_rnguniformminr%r&) rr6r8r'r(r)r*rngr0rs r2r<r<3sFB "BJ'''A "BJ'''A ;q!  DAq )   % %C  AsM400 {{{&&6 EKKK % % a7Nr4)F)rrr r r r )r5r)rrr6r7r8r9r r ) __future__rtypingrrnumpyrcontourpy._contourpyrr3r<r4r2rFs""""""%%%%%%%%5444444/4%%%%%RFIr4