o d@sddlZddlZddlmZmZejddZejddZejddZejd d Z ejdd gd d d Z ejddZ ejddZ ejddZ ejddZejddZejddZejddgd ddZejddd dd!dd"dggd#d$d%d&Zejddgd d'd(Zejddgd d)d*Zejddgd d+d,Zejd-d.gd d/d0Zejddgd d1d2Zejd3d4Zejd5efd6d7ZdS)8N)SeriesoptionscCt)z3A fixture providing the ExtensionDtype to validate.NotImplementedErrorrrd/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/extension/conftest.pydtype r cCr)z Length-100 array for this type. * data[0] and data[1] should both be non missing * data[0] and data[1] should not be equal rrrrrdatar cCr)z3Length-100 array in which all the elements are two.rrrrr data_for_twosr r cCr)zLength-2 array with [NA, Valid]rrrrr data_missing"r r)paramscCs |jdkr|S|jdkr|SdS)z5Parametrized fixture giving 'data' and 'data_missing'r rNparam)requestr rrrrall_data(s  rcsfdd}|S)a  Generate many datasets. Parameters ---------- data : fixture implementing `data` Returns ------- Callable[[int], Generator]: A callable that takes a `count` argument and returns a generator yielding `count` datasets. c3st|D]}VqdSN)range)count_r rrgenAs zdata_repeated..genr)r rrrr data_repeated1s rcCr)zn Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C rrrrrdata_for_sortingHr rcCr)z{ Length-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. rrrrrdata_missing_for_sortingSr rcCstjS)z Binary operator for comparing NA values. Should return a function of two arguments that returns True if both arguments are (scalar) NA for your type. By default, uses ``operator.is_`` )operatoris_rrrrna_cmp^s rcCdS)z6The scalar missing value for this type. Default 'None'Nrrrrrna_valuekr r!cCr)z Data for factorization, grouping, and unique tests. Expected to be like [B, B, NA, NA, A, A, B, C] Where A < B < C and NA is missing rrrrrdata_for_groupingqs r"TFcC|jS)z#Whether to box the data in a Seriesrrrrr box_in_series}sr%cCr Nrxrrrr*cCsdgt|Sr&)lenr(rrrr*scCstdgt|Sr&)rr,r(rrrr*scCs|Srrr(rrrr*r+)scalarlistseriesobject)ridscCr#)z, Functions to test groupby.apply(). rr$rrrgroupby_apply_ops r2cCr#)zU Boolean fixture to support Series and Series.to_frame() comparison testing. rr$rrras_framer3cCr#)zL Boolean fixture to support arr and Series(arr) comparison testing. rr$rrr as_seriesr4r5cCr#)zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. rr$rrr use_numpyr6ffillbfillcCr#)z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. rr$rrr fillna_methodr7r:cCr#)zR Boolean fixture to support ExtensionDtype _from_sequence method testing. rr$rrras_arrayr4r;cCs ttS)z A scalar that *cannot* be held by this ExtensionArray. The default should work for most subclasses, but is not guaranteed. If the array can hold any item (i.e. object dtype), then use pytest.skip. )r0__new__rrrrinvalid_scalars r=returncCstjjo tjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)rmode copy_on_write data_managerrrrrusing_copy_on_writesrC)rpytestpandasrrfixturer r r rrrrrrr!r"r%r2r3r5r6r:r;r=boolrCrrrrsd