h<SrSSKrSSKrSSKJr SSKrSSKJ r SSK J r SSK Jr SSKJr \ R"rSS\S\4S jjr\R*"S S /S 9S 5r\R*S5r\R*S5r\R*S5r\R*S5r\R*S5r\R*S5r\R*S5r\R*S5r\R*S5r\R@RCS5r""SS\RF5r$"SS\RJ5r&g)a This file contains a minimal set of tests for compliance with the extension array interface test suite, and should contain no other tests. The test suite for the full functionality of the array is located in `pandas/tests/arrays/`. The tests in this file are inherited from the BaseExtensionTests, and only minimal tweaks should be applied to get the tests passing (by overwriting a parent method). Additional tests should either be added to one of the BaseExtensionTests classes (if they are relevant for the extension interface for all dtypes), or be added to the array-specific tests in `pandas/tests/arrays/`. Note: we do not bother with base.BaseIndexTests because NumpyExtensionArray will never be held in an Index. N) NumpyEADtype)is_object_dtype)NumpyExtensionArray)baseattrobjcrUS:Xa[USS5n[USS5n[U[5(a1[U[5(dURUR5nOE[U[5(a0[U[5(dURUR5n[ XX#5 g)z] patch tm.assert_attr_equal so NumpyEADtype("object") is closed enough to np.dtype("object") dtypeN)getattr isinstancerastype numpy_dtypeorig_assert_attr_equal)rleftrightrlattrrattrs eC:\Users\psant\Downloads\Noiseback\school\venv\Lib\site-packages\pandas/tests/extension/test_numpy.py_assert_attr_equalr s  wgt,w- e\ * *:e\3R3R;;u001D | , ,Z|5T5TLL!2!23E4u2floatobject)paramscT[[R"UR55$N)rnpr param)requests rr r 0s / 00rc## UR5nUR[SS5 UR[RS[ 5 Sv SSS5 g!,(df  g=f7f)a A monkeypatch to tells pandas to let us in. By default, passing a NumpyExtensionArray to an index / series / frame constructor will unbox that NumpyExtensionArray to an ndarray, and treat it as a non-EA column. We don't want people using EAs without reason. The mechanism for this is a check against ABCNumpyExtensionArray in each constructor. But, for testing, we need to allow them in pandas. So we patch the _typ of NumpyExtensionArray, so that we evade the ABCNumpyExtensionArray check. _typ extensionassert_attr_equalN)contextsetattrrtm assertersr) monkeypatchms rallow_in_pandasr)5sJ"    ! %v{; ",, 35GH    sA.AA A. A+'A.cURS:Xa:[R"[S5Vs/sHo"4PM sn5R$[ [ R"SSURS95$s snf)Nrder ) rpdSeriesrangearrayrrarange_dtype)r)r is rdatar6LsY H$yyc 3 1$ 34::: ryyCu||D EE4s A4cURS:Xa2[[R"[RS/[ S95$[[R"[RS/55$)Nrr,r.g?rrrr2nanrr)r s r data_missingr<SsJ H$"288RVVTN&#IJJ rxx 6 77rc SnU$)Nch[R"U5=(a [R"U5$r)risnan)abs rcmpna_cmp..cmp\sxx{*rxx{*r)rBs rna_cmprEZs+ JrcURS:Xa'[[R"/SQ[S9SS5$[[R"/SQ55$)z]Length-3 array with a known sort order. This should be three items [B, C, A] with A < B < C r)rDr8r.r,N)r,rHr)rrrr2rr;s rdata_for_sortingrKbsH H$#288,B&#QRSRT#UVV rxx 2 33rcURS:Xa3[[R"S[RS/[ S95$[[R"S[RS/55$)zjLength-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. rr8)rr.r,rr9r;s rdata_missing_for_sortingrMpsQ H$"288T2664,@#OPP rxxBFFA7 88rc URS:XaSup#nO[R"S5up#n[[R"X3[R [R X"X4/URS95$)zData 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 r)r8rGrIrJr.)rrr3rr2r:)r)r r@rAcs rdata_for_groupingrP|s` H$"a))A,a  !a35;L;LM rcURS:Xa[R"US35 [R"S5S-n[ R "XS9$)NOz is not a numeric dtyper+rHr.)kindpytestskipronesr_from_sequence)r arrs r data_for_twosrYsE zzS ug456 ''#, C  - -c ??rcnUS:Xa/[RRSS9nURU5 gg)ax Tests for NumpyExtensionArray with nested data. Users typically won't create these objects via `pd.array`, but they can show up through `.array` on a Series with nested data. Many of the base tests fail, as they aren't appropriate for nested data. This fixture allows these tests to be skipped when used as a usefixtures marker to either an individual test or a test class. rzFails for object dtypereasonN)rTmarkxfail applymarker)r rr]s rskip_numpy_objectr`s7 {{  (@ AD!rr`c ^\rSrSr\R R SS9S5r\U4Sj5r U4Sjr U4Sjr \U4Sj5r \U4S j5r \U4S j5r\U4S j5r\U4S j5r\R R#S S9U4Sj5rU4Sjr\U4Sj5rSrSrSrSrU4SjrSrU4SjrU4SjrU4SjrS\R>S\ S\!4Sjr"S\R>S\ S\!4Sjr#\R R S5\R RISSS/5S 55r%\U4S!j5r&\U4S"j5r'\U4S#j5r(\U4S$j5r)\\R RIS%S&S/5U4S'j55r*\U4S(j5r+\\R RIS)\,RZ"/S*Q5\RZ"/S*QS+S,9/S-S./S/9U4S0j55r.\\R RIS1/S2Q\RZ"/S2QS3S,9\,RZ"/S2Q5//S4QS/9U4S5j55r/\R RIS6S7S8S9\R`/S4\Rb"S7S8S9\R`/S\R R"S:9\RZ"S7S8S9\R`/S3S,9S4\RZ"S7S8S9\R`/S3S,9S4//S;QS/9U4S<j5r2\U4S=j5r3\U4S>j5r4S?r5\R R#S@S9U4SAj5r6\\R RISBSCSD/5U4SEj55r7SFr8U=r9$)GTestNumpyExtensionArrayzWe don't register our dtyper[cgrrD)selfr6s rtest_from_dtype'TestNumpyExtensionArray.test_from_dtype rc$>[TU]X5 gr)super)test_series_constructor_scalar_with_index)rer6r __class__s rrkATestNumpyExtensionArray.test_series_constructor_scalar_with_indexs 9$Frc>URRS:XaCUR[RR SURR3S95 [ TU]U5 g)Nrz:NumpyExtensionArray expectedly clashes with a NumPy name: r[)r rr_rTr]r^rjtest_check_dtype)rer6rusing_infer_stringrls rro(TestNumpyExtensionArray.test_check_dtypesc :: ! !X -    !!##'::#9#9":<"   &rcj>URS:Xa[U5(deg[TU] U5 g)Nr)rrrjtest_is_not_object_type)rer rrls rrs/TestNumpyExtensionArray.test_is_not_object_types1    (#5)) )) G +E 2rc$>[TU]U5 gr)rjtest_getitem_scalarrer6rls rrv+TestNumpyExtensionArray.test_getitem_scalars #D)rc$>[TU]U5 gr)rjtest_shift_fill_valuerws rrz-TestNumpyExtensionArray.test_shift_fill_values %d+rc$>[TU]U5 gr)rjtest_fillna_copy_framerer<rls rr}.TestNumpyExtensionArray.test_fillna_copy_frames &|4rc$>[TU]U5 gr)rjtest_fillna_copy_seriesr~s rr/TestNumpyExtensionArray.test_fillna_copy_seriess ' 5rc$>[TU]X5 gr)rjtest_searchsorted)rerK as_seriesrls rr)TestNumpyExtensionArray.test_searchsorteds !"2>rz*NumpyExtensionArray.diff may fail on dtypec">[TU]X5$r)rj test_diff)rer6periodsrls rr!TestNumpyExtensionArray.test_diffsw //rc>URR[:Xa.[RR SS9nUR U5 [TU]!U5 g)Nz$Dimension mismatch in np.concatenater[) r rrrTr]r^r_rj test_insert)rer6rr]rls rr#TestNumpyExtensionArray.test_insertsH :: ! !V +;;$$,R$SD    % D!rc$>[TU]X5 gr)rjtest_insert_invalidrer6invalid_scalarrls rr+TestNumpyExtensionArray.test_insert_invalids #D9rNct>SnURRS:Xa[nX l[TU]U5 gNrR)r rS TypeError divmod_excrj test_divmod)rer6rrls rr#TestNumpyExtensionArray.test_divmods/ ::??c !"J$ D!rc[R"U5nSnURRS:Xa [nX0lUR U[U5 gr)r/r0r rSrr_check_divmod_opdivmod)rer6serexcs rtest_divmod_series_array0TestNumpyExtensionArray.test_divmod_series_arrays@iio ::??c !C!O c640rc>UnSnURR[:XaDUS;a8[RR SS9nUR RU5 [nXPl [TU]-X5 gN)__mul____rmul__z9the Series.combine step raises but not the Series method.r[) r rrrTr]r^node add_markerrseries_scalar_excrjtest_arith_series_with_scalar)rer6all_arithmetic_operatorsropnamerr]rls rr5TestNumpyExtensionArray.test_arith_series_with_scalarsr)  :: ! !V +00{{((V) ''- ) !2 -dMrc>UnSnURR[:Xa US;a[nX@l[ TU]X5 g)N)__add____radd__)r rrrseries_array_excrjtest_arith_series_with_array)rer6rrrrls rr4TestNumpyExtensionArray.test_arith_series_with_arraysA) :: ! !V +>U0U(  0 ,TLrc>UnSnURR[:XaDUS;a8[RR SS9nUR RU5 [nXPl [TU]-X5 gr) r rrrTr]r^rrrframe_scalar_excrjtest_arith_frame_with_scalar)rer6rrrrr]rls rr4TestNumpyExtensionArray.test_arith_frame_with_scalarsr) :: ! !V +00{{((V) ''-(  0 ,TLrrop_namereturncBURRS:XaUS;$g)NrR)summinmaxanyallT)r rS)rerrs r_supports_reduction+TestNumpyExtensionArray._supports_reduction&s! 99>>S AA Arskipnac[X5nURRnURU5n[Xb5nUS:XaU"5nU"5n O U"US9nU"US9n [R "X5 g)Ncount)r)r r rr r%assert_almost_equal) rerrrres_op cmp_dtypealtexp_opresultexpecteds r check_reduce$TestNumpyExtensionArray.check_reduce+sm&II)) jj#& g XFxH6*FV,H v0rzTODO: tests not written yetTFcgrrD)rer6all_numeric_reductionsrs rtest_reduce_frame)TestNumpyExtensionArray.test_reduce_frame;rhrc$>[TU]U5 gr)rjtest_fillna_seriesr~s rr*TestNumpyExtensionArray.test_fillna_series@s "<0rc$>[TU]U5 gr)rjtest_fillna_framer~s rr)TestNumpyExtensionArray.test_fillna_frameEs !,/rc$>[TU]X5 gr)rjtest_setitem_invalidrs rr,TestNumpyExtensionArray.test_setitem_invalidJs $T:rc$>[TU]X5 gr)rj test_setitem_sequence_broadcastsrer6 box_in_seriesrls rr8TestNumpyExtensionArray.test_setitem_sequence_broadcastsOs 0Ersetterlocc$>[TU]X5 gr)rjtest_setitem_mask_broadcast)rer6rrls rr3TestNumpyExtensionArray.test_setitem_mask_broadcastUs +D9rc$>[TU]U5 gr)rj&test_setitem_scalar_key_sequence_raiserws rr>TestNumpyExtensionArray.test_setitem_scalar_key_sequence_raise\s 6t[TU]XU5 gr)rjtest_setitem_mask)rer6rrrls rr)TestNumpyExtensionArray.test_setitem_maskds !$m[TU]XU5 gr)rjtest_setitem_integer_arrayrer6rrrls rr2TestNumpyExtensionArray.test_setitem_integer_arrayps *4mDrzidx, box_in_seriesrr,rH)marks)z list-Falsez list-Truezinteger-array-Falsezinteger-array-Truec&>[TU]XU5 gr)rj(test_setitem_integer_with_missing_raisesrs rr@TestNumpyExtensionArray.test_setitem_integer_with_missing_raisesys 8MRrc$>[TU]X5 gr)rjtest_setitem_slicers rr*TestNumpyExtensionArray.test_setitem_slices "47rc$>[TU]U5 gr)rjtest_setitem_loc_iloc_slicerws rr3TestNumpyExtensionArray.test_setitem_loc_iloc_slices +D1rc[R"S[R"U505=p4[R"URS9nU"U5nUSURUS4'UR R [:waJ[U[5(aU[S5:wa&[R"SUR505n[R"XTSS9 g)Nr6)indexF)check_column_type) r/ DataFramer0rrr rrr sliceto_numpyr%assert_frame_equal)rer6 full_indexerdfrrkeys r,test_setitem_with_expansion_dataframe_columnDTestNumpyExtensionArray.test_setitem_with_expansion_dataframe_columns fbiio%>??BHH-2"$V* 3; :: ! !V +c5))SE$K-?<<(AB f%HrzNumpyEADtype is unpackedc$>[TU]U5 gr)rj#test_index_from_listlike_with_dtyperws rr ;TestNumpyExtensionArray.test_index_from_listlike_with_dtypes 3D9renginerOpythonc&>[TU]XU5 gr)rj test_EA_types)rer r6rrls rr%TestNumpyExtensionArray.test_EA_typess fG4r)rrrr):__name__ __module__ __qualname____firstlineno__rTr]rUrf skip_nestedrkrorsrvrzr}rrr^rrrrrrrrrrrrr/r0strboolrr parametrizerrrrrrrrr2rrNArrrrrr r__static_attributes__ __classcell__)rls@rrbrbs$ [[:; < GG'3**,,5566??  [[JK0L0"::J"1 NM Mryy34 1 1C11  [[34 [[Xe}5 65 1100;;FF  [[Xt}5:6: == [[ HH5 6 HH5Y G O , == [[ BHHYg68KL 4 E  E [[Aruu u % LL!Q255)4v{{7H7H I XXq!Q&g 6 > XXq!Q&g 6 >  U S S8822I" [[89::: [[XX7585rrbc\rSrSrSrg) Test2DCompatirDN)rrrrrrDrrrrsrr) Attributes)'__doc__numpyrrTpandas.core.dtypes.dtypesrpandasr/pandas._testing_testingr%pandas.api.typesrpandas.core.arrays.numpy_rpandas.tests.extensionrr"rrrfixturer r)r6r<rErKrMrPrYr`r] usefixturesrExtensionTestsrbNDArrayBacked2DTestsrrDrrr,s" 2,9'-- 3S 3C 3 *+1,1,FF 88  4 499   @@ " "kk%%&9: ~5d11~5B 4,, r