a j=ic@sddlZddlZddlmZejddZejddZejddZejd d Zejdd gd d d Z ejddZ ejddZ ejddZ ejddZ ejddZejddZejddgd ddZejddd dd!dd"dggd#d$d%d&Zejddgd d'd(Zejddgd d)d*Zejddgd d+d,Zejd-d.gd d/d0Zejddgd d1d2Zejd3d4ZdS)5N)SeriescCstdS)z3A fixture providing the ExtensionDtype to validate.NNotImplementedErrorrrp/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/extension/conftest.pydtypesrcCstdS)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 NrrrrrdatasrcCstdS)z3Length-100 array in which all the elements are two.Nrrrrr data_for_twossr cCstdS)zLength-2 array with [NA, Valid]Nrrrrr data_missingsr )paramscCs |jdkr|S|jdkr|SdS)z5Parametrized fixture giving 'data' and 'data_missing'rr Nparam)requestrr rrrall_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_rrrgen>s zdata_repeated..genr)rrrrr data_repeated.s rcCstdS)zn Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C Nrrrrrdata_for_sortingEsrcCstdS)z{ Length-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. Nrrrrrdata_missing_for_sortingPsrcCstjS)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 rcCsdS)z6The scalar missing value for this type. Default 'None'Nrrrrrna_valuehsrcCstdS)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 Nrrrrrdata_for_groupingns rTFcCs|jS)z#Whether to box the data in a Seriesr rrrr box_in_serieszsrcCsdSNrxrrrr$cCsdgt|Sr )lenr"rrrr$r%cCstdgt|Sr )rr&r"rrrr$r%cCs|Srrr"rrrr$r%)ZscalarlistZseriesobject)r ZidscCs|jS)z, Functions to test groupby.apply(). r rrrrgroupby_apply_ops r)cCs|jS)zU Boolean fixture to support Series and Series.to_frame() comparison testing. r rrrras_framesr*cCs|jS)zL Boolean fixture to support arr and Series(arr) comparison testing. r rrrr as_seriessr+cCs|jS)zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. r rrrr use_numpysr,ZffillZbfillcCs|jS)z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. r rrrr fillna_methodsr-cCs|jS)zR Boolean fixture to support ExtensionDtype _from_sequence method testing. r rrrras_arraysr.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. )r(__new__rrrrinvalid_scalars r0)rZpytestZpandasrZfixturerrr r rrrrrrrrr)r*r+r,r-r.r0rrrrs^