U 6c5@s ddlZd ddZd ddZdS) NFc Cs|\}}tj|tjd}tj|tjd}t||\}}|d}|d}tdddddg}tddgd dgd d gdd gd dgg} tddddd g} t|} tt|D]T} | || t||| | dfd||| | dfd | | d7} q|rvt ||ddd||ddd dk||ddd||ddddk} tj j | | d} ||| 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``. Zdtypeg?gg?ggffffff?g?g?g333333?g?g?g?g?rg{Gz?g?g{Gz?mask) nparangefloat64meshgridZasarrayZ zeros_likerangelenexp logical_ormaarray)shapeZ want_masknynxxyZxscaleZyscaleampZmidwidthzirr7/tmp/pip-unpacked-wheel-9ja4spaj/contourpy/util/data.pysimples& ( R**rrc Cs|\}}tj|tjd}tj|tjd}t||\}}tj|}|j|d}|dkrt|d}|j|d|k} tjj || d}|||fS)aReturn random test data.. 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) rr r r randomZ default_rnguniformminrr) rseedZ mask_fractionrrrrrngrrrrrr *s    r )F)rr)Znumpyrrr rrrrs &