d~ddlZddlZddlmZmZejdZejdZejdZejdZ ejddg d Z ejd Z ejd Z ejd Z ejdZejdZejdZejddg dZejddddggddZejddg dZejddg dZejddg dZejddg d Zejddg d!Zejd"Zejd#efd$ZdS)%N)Seriesoptionsct)z3A fixture providing the ExtensionDtype to validate.NotImplementedError[/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/extension/conftest.pydtyper  r ct)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 rrr r datar  r ct)z3Length-100 array in which all the elements are two.rrr r data_for_twosrr r ct)zLength-2 array with [NA, Valid]rrr r data_missingr"r r rr)paramsc:|jdkr|S|jdkr|SdS)z5Parametrized fixture giving 'data' and 'data_missing'rrNparam)requestrrs r all_datar(s1} . ( ( ) (r cfd}|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. c38Kt|D]}VdSN)range)count_rs r genzdata_repeated..genAs1u  AJJJJ  r r)rr s` r data_repeatedr!1s#  Jr ct)zn Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C rrr r data_for_sortingr#Hrr ct)z{ Length-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. rrr r data_missing_for_sortingr%Srr ctjS)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_rr r na_cmpr)^s  <r cdS)z6The scalar missing value for this type. Default 'None'Nrrr r na_valuer+ks  4r ct)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 rrr r data_for_groupingr-qs  r TFc|jS)z#Whether to box the data in a Seriesrrs r box_in_seriesr0}s  =r cdSNrxs r r6!r c(dgt|zSr2)lenr4s r r6r6s1#A,r cBtdgt|zSr2)rr9r4s r r6r6s&!s1vv&&r c|Srrr4s r r6r6r7r )scalarlistseriesobject)ridsc|jS)z, Functions to test groupby.apply(). rr/s r groupby_apply_oprBs  =r c|jS)zU Boolean fixture to support Series and Series.to_frame() comparison testing. rr/s r as_framerD =r c|jS)zL Boolean fixture to support arr and Series(arr) comparison testing. rr/s r as_seriesrGrEr c|jS)zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. rr/s r use_numpyrI =r ffillbfillc|jS)z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. rr/s r fillna_methodrNrJr c|jS)zR Boolean fixture to support ExtensionDtype _from_sequence method testing. rr/s r as_arrayrPrEr c@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__)rs r invalid_scalarrSs >>& ! !!r returncNtjjotjjdkS)z7 Fixture to check if Copy-on-Write is enabled. block)rmode copy_on_write data_managerrr r using_copy_on_writerZs < % N',*Cw*NNr )r'pytestpandasrrfixturer rrrrr!r#r%r)r+r-r0rBrDrGrIrNrPrSboolrZrr r r_sZ     /00010,    e}%%%&%  &&   /..e}%%%&%e}%%%&%e}%%%&%)***+*e}%%%&%"""OTOOOOOOr