a Pf@sddlZddlZddlmZddlmZmZejddZejddZ ejdd Z ejd d Z ejdd gd d dZ ejddZ ejddZejddZejddZejddZejddZejddgd ddZejdd d!d d"d d#d ggd$d%d&d'Zejddgd d(d)Zejddgd d*d+Zejddgd d,d-Zejd.d/gd d0d1Zejddgd d2d3Zejd4d5Zejed6d7d8ZdS)9N) _get_option)SeriesoptionscCstdS)z3A fixture providing the ExtensionDtype to validate.NNotImplementedErrorrrX/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/extension/conftest.pydtype sr cCstdS)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 Nrrrrrdatasr cCs(|js |jdks t|dtdS)z Length-100 array in which all the elements are two. Call pytest.skip in your fixture if the dtype does not support divmod. mz is not a numeric dtypeN)Z _is_numerickindpytestskiprr rrr data_for_twossrcCstdS)zLength-2 array with [NA, Valid]Nrrrrr data_missing-sr)paramscCs |jdkr|S|jdkr|SdS)z5Parametrized fixture giving 'data' and 'data_missing'r rNparam)requestr rrrrall_data3s  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 rrgenLs zdata_repeated..genr)r rrrr data_repeated<s rcCstdS)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 Nrrrrrdata_for_sortingSs rcCstdS)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_sortingasrcCstjS)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_cmpls r"cCs|jS)z The scalar missing value for this type. Default dtype.na_value. TODO: can be removed in 3.x (see https://github.com/pandas-dev/pandas/pull/54930) )na_valuerrrrr#ysr#cCstdS)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. Nrrrrrdata_for_groupings r$TFcCs|jS)z#Whether to box the data in a Seriesrrrrr box_in_seriessr&cCsdSNrxrrrr+cCsdgt|Sr')lenr)rrrr+r,cCstdgt|Sr')rr-r)rrrr+r,cCs|Srrr)rrrr+r,)ZscalarlistZseriesobject)rZidscCs|jS)z, Functions to test groupby.apply(). rr%rrrgroupby_apply_ops r0cCs|jS)zU Boolean fixture to support Series and Series.to_frame() comparison testing. rr%rrras_framesr1cCs|jS)zL Boolean fixture to support arr and Series(arr) comparison testing. rr%rrr as_seriessr2cCs|jS)zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. rr%rrr use_numpysr3ZffillZbfillcCs|jS)z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. rr%rrr fillna_methodsr4cCs|jS)zR Boolean fixture to support ExtensionDtype _from_sequence method testing. rr%rrras_arraysr5cCs 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 r7)returncCstjjduotddddkS)z7 Fixture to check if Copy-on-Write is enabled. Tzmode.data_manager)Zsilentblock)rmodeZ copy_on_writerrrrrusing_copy_on_writes r;)r r Zpandas._config.configrZpandasrrZfixturer r rrrrrrr"r#r$r&r0r1r2r3r4r5r7boolr;rrrrsd