LVfddlZddlZddlmZmZej dZej dZej dZej dZ ej ddg d Z ej d Z ej d Z ej d Z ej dZej dZej dZej ddg dZej ddddggddZej ddg dZej ddg dZej ddg dZej ddg d Zej ddg d!Zej d"Zej d#efd$Zy)%N)Seriesoptionsct)z3A fixture providing the ExtensionDtype to validate.NotImplementedErrorh/var/www/html/software/conda/envs/catlas/lib/python3.12/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 cn|js$|jdk(stjdt)z Length-100 array in which all the elements are two. Call pytest.skip in your fixture if the dtype does not support divmod. mzNot a numeric dtype) _is_numerickindpytestskiprr s r data_for_twosrs-   s!2  )* r ct)zLength-2 array with [NA, Valid]rrr r data_missingr+r r rr)paramscH|jdk(r|S|jdk(r|Sy)z5Parametrized fixture giving 'data' and 'data_missing'rrNparam)requestrrs r all_datar1s,}} . ( )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. c36Kt|D]}ywN)range)count_rs r genzdata_repeated..genJsu AJ sr)rr&s` r data_repeatedr':s  Jr ct)z Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C For boolean dtypes (for which there are only 2 values available), set B=C=True rrr r data_for_sortingr)Qs  r 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+_rr c"tjS)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/js <<r c|jS)z>The scalar missing value for this type. Default dtype.na_value)na_valuers r r1r1ws >>r 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. If a dtype has _is_boolean = True, i.e. only 2 unique non-NA entries, then set C=B. rrr r data_for_groupingr3}s  r TFc|jS)z#Whether to box the data in a Seriesrrs r box_in_seriesr6s ==r cyNrxs r r<sr c dgt|zSr8)lenr:s r r<r<s1#A,r c2tdgt|zSr8)rr>r:s r r<r<s&!s1v&r c|Sr"rr:s r r<r<s!r )scalarlistseriesobject)ridsc|jS)z, Functions to test groupby.apply(). rr5s r groupby_apply_oprGs ==r c|jS)zU Boolean fixture to support Series and Series.to_frame() comparison testing. rr5s r as_framerI ==r c|jS)zL Boolean fixture to support arr and Series(arr) comparison testing. rr5s r as_seriesrLrJr c|jS)zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. rr5s r use_numpyrN ==r ffillbfillc|jS)z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. rr5s r fillna_methodrSrOr c|jS)zR Boolean fixture to support ExtensionDtype _from_sequence method testing. rr5s r as_arrayrUrJr c4tjtS)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. )rD__new__)rs r invalid_scalarrXs >>& !!r returncttjjxrtjjdk(S)z7 Fixture to check if Copy-on-Write is enabled. block)rmode copy_on_write data_managerrr r using_copy_on_writer_s( << % % N',,*C*Cw*NNr )r-rpandasrrfixturer rrrrr'r)r+r/r1r3r6rGrIrLrNrSrUrXboolr_rr r rcsf      /01,       e}%& &   /e}%&e}%&e}%&)*+e}%&""OTOOr