o g@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 d1d2ZdS)3N)SeriescCt)z3A fixture providing the ExtensionDtype to validate.NotImplementedErrorrr]/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/pandas/tests/extension/conftest.pydtypercCr)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_missingr r )paramscCs |jdkr|S|jdkr|SdS)z5Parametrized fixture giving 'data' and 'data_missing'r r Nparam)requestr r 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_r rrgen>s zdata_repeated..genr)r rrrr data_repeated.s rcCr)zn Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C rrrrrdata_for_sortingEr 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_sortingPr 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_valuehr 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_groupingns r!TFcC|jS)z#Whether to box the data in a Seriesrrrrr box_in_serieszsr$cCrNrxrrrr)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 r1cCr")zU Boolean fixture to support Series and Series.to_frame() comparison testing. rr#rrras_framer2cCr")zL Boolean fixture to support arr and Series(arr) comparison testing. rr#rrr as_seriesr3r4cCr")zd Boolean fixture to support comparison testing of ExtensionDtype array and numpy array. rr#rrr use_numpyr5ffillbfillcCr")z{ Parametrized fixture giving method parameters 'ffill' and 'bfill' for Series.fillna(method=) testing. rr#rrr fillna_methodr6r9cCr")zR Boolean fixture to support ExtensionDtype _from_sequence method testing. rr#rrras_arrayr3r:)rpytestpandasrfixturerr r r rrrrrr r!r$r1r2r4r5r9r:rrrrs\