!h<VdZddlZddlZddlmZddlZddlm Z ddl m Z ddl mZddlmZe jZdded efd Zejd d g dZejdZejdZejdZejdZejdZejdZejdZejdZejdZej !dZ"Gddej#Z$Gddej%Z&dS)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)base Attributesattrobjc|dkrt|dd}t|dd}t|tr0t|ts||j}nDt|tr/t|ts||j}t ||||dS)zi patch tm.assert_attr_equal so NumpyEADtype("object") is closed enough to np.dtype("object") dtypeN)getattr isinstancerastype numpy_dtypeorig_assert_attr_equal)rleftrightr lattrrattrs [c:\xampp_lite_8_4\www\timesheet\venv\Lib\site-packages\pandas/tests/extension/test_numpy.py_assert_attr_equalr s  wgt,,w-- e\ * * 4:e\3R3R 4;;u011DD | , , 4Z|5T5T 4LL!233E4uc22222floatobject)paramscNttj|jSN)rnpr param)requests rr r 0s // 0 00rc#K|5}|tdd|tjdt dVddddS#1swxYwYdS)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*5s"     ! %v{;;; ", 35GHHH sAA++A/2A/c|jdkr0tjdtdDjSt t jdd|jS)Nrcg|]}|fSr-).0is r zdata..Os33311$333rder ) rpdSeriesrangearrayrrarange_dtyper*r s rdatar<LsY H$$y33c 33344:: ryCu|DDD E EErc|jdkr4ttjtjdgt SttjtjdgS)Nrr2r4g?rrrr8nanrr;s r data_missingrASsQ H$$"28RVTN&#I#I#IJJJ rx 66 7 77rc d}|S)NcRtj|otj|Sr)risnan)abs rcmpzna_cmp..cmp\sx{{*rx{{*rr-)rGs rna_cmprHZs+++ Jrc|jdkr2ttjgdtddSttjgdS)ziLength-3 array with a known sort order. This should be three items [B, C, A] with A < B < C r)r-r>r4r2N)r2rKr)rrrr8rr;s rdata_for_sortingrNbs` H$$#28,B,B,B&#Q#Q#QRSRTRT#UVVV rx 22 3 33rc|jdkr5ttjdtjdgt SttjdtjdgS)zvLength-3 array with a known sort order. This should be three items [B, NA, A] with A < B and NA missing. rr>)rr4r2rr?r;s rdata_missing_for_sortingrPpsX H$$"28T264,@#O#O#OPPP rxBFA77 8 88rc |jdkrd\}}}ntjd\}}}ttj||tjtj||||g|jS)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)r>rJrLrMr4)rrr9rr8r@)r*r rErFcs rdata_for_groupingrS|sn H$$"1aa)A,,1a  !Q1a35;LMMM  rc|jdkrtj|dtjddz}t j||S)NOz is not a numeric dtyper1rKr4)kindpytestskipronesr_from_sequence)r arrs r data_for_twosr\sP zS u555666 '#,, C  -c ? ? ??rc|dkr7tjd}||dSdS)a 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)rWmarkxfail applymarker)r rr`s rskip_numpy_objectrcsJ {  (@ AAD!!!!!rrcc eZdZejddZefdZfdZ fdZ efdZ efdZ efd Z efd Zefd Zejd fd ZfdZefdZdZdZdZdZfdZdZfdZfdZfdZdejdede fdZ!dejdede fdZ"ejdej#dddgdZ$efd Z%efd!Z&efd"Z'efd#Z(eej#d$d%dgfd&Z)efd'Z*eej#d(e+j,gd)ej,gd)d*+gd,d-g.fd/Z-eej#d0gd1ej,gd1d2+e+j,gd1ggd3.fd4Z.ej#d5d6d7d8ej/gdfej0d6d7d8ej/gdejj9ej,d6d7d8ej/gd2+dfej,d6d7d8ej/gd2+dfggd:.fd;Z1efd<Z2efd=Z3d>Z4ejd?fd@Z5eej#dAdBdCgfdDZ6xZ7S)ETestNumpyExtensionArrayzWe don't register our dtyper^cdSrr-)selfr<s rtest_from_dtypez'TestNumpyExtensionArray.test_from_dtype  rcLt||dSr)super)test_series_constructor_scalar_with_index)rgr<r __class__s rrlzATestNumpyExtensionArray.test_series_constructor_scalar_with_indexs% 99$FFFFFrc|jjdkr@|tjd|jjt |dS)Nrz:NumpyExtensionArray expectedly clashes with a NumPy name: r^)r rrbrWr`rarktest_check_dtype)rgr<rusing_infer_stringrms rroz(TestNumpyExtensionArray.test_check_dtypes| : !X - -    !!<#':#9<<"      &&&&&rc|jdkrt|sJdSt|dS)Nr)rrrktest_is_not_object_type)rgr rrms rrrz/TestNumpyExtensionArray.test_is_not_object_typesK   ( (#5)) ) )) ) ) GG + +E 2 2 2 2 2rcJt|dSr)rktest_getitem_scalarrgr<rms rrtz+TestNumpyExtensionArray.test_getitem_scalars# ##D)))))rcJt|dSr)rktest_shift_fill_valuerus rrwz-TestNumpyExtensionArray.test_shift_fill_values# %%d+++++rcJt|dSr)rktest_fillna_copy_framergrArms rryz.TestNumpyExtensionArray.test_fillna_copy_frames# &&|44444rcJt|dSr)rktest_fillna_copy_seriesrzs rr|z/TestNumpyExtensionArray.test_fillna_copy_seriess# '' 55555rcLt||dSr)rktest_searchsorted)rgrN as_seriesrms rr~z)TestNumpyExtensionArray.test_searchsorteds& !!"2I>>>>>rz*NumpyExtensionArray.diff may fail on dtypecHt||Sr)rk test_diff)rgr<periodsrms rrz!TestNumpyExtensionArray.test_diffsww  w///rc|jjtkr5tjd}||t|dS)Nz$Dimension mismatch in np.concatenater^) r rrrWr`rarbrk test_insert)rgr<rr`rms rrz#TestNumpyExtensionArray.test_inserts^ : !V + +;$$,R$SSD    % % % D!!!!!rcLt||dSr)rktest_insert_invalidrgr<invalid_scalarrms rrz+TestNumpyExtensionArray.test_insert_invalids% ##D.99999rNcd}|jjdkrt}||_t |dSNrU)r rV TypeError divmod_excrk test_divmod)rgr<rrms rrz#TestNumpyExtensionArray.test_divmodsA :?c ! !"J$ D!!!!!rctj|}d}|jjdkrt}||_||t|dSr)r5r6r rVrr_check_divmod_opdivmod)rgr<serexcs rtest_divmod_series_arrayz0TestNumpyExtensionArray.test_divmod_series_arraysMioo :?c ! !C!DO c6400000rc|}d}|jjtkrE|dvr:tjd}|j|t}||_ t ||dSN)__mul____rmul__z9the Series.combine step raises but not the Series method.r^) r rrrWr`ranode add_markerrseries_scalar_excrktest_arith_series_with_scalar)rgr<all_arithmetic_operatorsropnamerr`rms rrz5TestNumpyExtensionArray.test_arith_series_with_scalars)  : !V + +000{((V) ''--- ) !2 --d4LMMMMMrc|}d}|jjtkr |dvrt}||_t ||dS)N)__add____radd__)r rrrseries_array_excrktest_arith_series_with_array)rgr<rrrrms rrz4TestNumpyExtensionArray.test_arith_series_with_arraysX) : !V + +>U0U0U(  0 ,,T3KLLLLLrc|}d}|jjtkrE|dvr:tjd}|j|t}||_ t ||dSr) r rrrWr`rarrrframe_scalar_excrktest_arith_frame_with_scalar)rgr<rrrrr`rms rrz4TestNumpyExtensionArray.test_arith_frame_with_scalars) : !V + +000{((V) ''---(  0 ,,T3KLLLLLrrop_namereturnc.|jjdkr|dvSdS)NrU)summinmaxanyallT)r rV)rgrrs r_supports_reductionz+TestNumpyExtensionArray._supports_reduction&s" 9>S AA Atrskipnact||}|jj}||}t||}|dkr|}|} n||}||} t j|| dS)Ncount)r)r r rrr&assert_almost_equal) rgrrrres_op cmp_dtypealtexp_opresultexpecteds r check_reducez$TestNumpyExtensionArray.check_reduce+sg&&I) jj##g&& g  VXXFvxxHHV6***FvV,,,H vx00000rzTODO: tests not written yetTFcdSrr-)rgr<all_numeric_reductionsrs rtest_reduce_framez)TestNumpyExtensionArray.test_reduce_frame;rircJt|dSr)rktest_fillna_seriesrzs rrz*TestNumpyExtensionArray.test_fillna_series@s# ""<00000rcJt|dSr)rktest_fillna_framerzs rrz)TestNumpyExtensionArray.test_fillna_frameEs# !!,/////rcLt||dSr)rktest_setitem_invalidrs rrz,TestNumpyExtensionArray.test_setitem_invalidJs% $$T>:::::rcLt||dSr)rk test_setitem_sequence_broadcastsrgr< box_in_seriesrms rrz8TestNumpyExtensionArray.test_setitem_sequence_broadcastsOs% 00}EEEEErsetterloccLt||dSr)rktest_setitem_mask_broadcast)rgr<rrms rrz3TestNumpyExtensionArray.test_setitem_mask_broadcastUs% ++D&99999rcJt|dSr)rk&test_setitem_scalar_key_sequence_raiserus rrz>TestNumpyExtensionArray.test_setitem_scalar_key_sequence_raise\s# 66t<<<<???XBH---l2"$V* 3; : !V + +c5)) CSE$KK-?-?<(ABB fh%HHHHHHrzNumpyEADtype is unpackedcJt|dSr)rk#test_index_from_listlike_with_dtyperus rrz;TestNumpyExtensionArray.test_index_from_listlike_with_dtypes! 33D99999renginerRpythoncNt|||dSr)rk test_EA_types)rgrr<rrms rrz%TestNumpyExtensionArray.test_EA_typess' fdG44444r)8__name__ __module__ __qualname__rWr`rXrh skip_nestedrlrorrrtrwryr|r~rarrrrrrrrrrrrr5r6strboolrr parametrizerrrrrrrrr8rrNArrrrrrr __classcell__)rms@rreres [:;;  <; GGGG[G'''''33333****[*,,,,[,5555[56666[6????[?  [JKK0000LK0"""""::::[:J"""""111 N N N N NMMMMM M M M M Mry34 1 1C11111  [344 [Xe}55  6554 1111[10000[0;;;;[;FFFF[F  [Xt}55::::65[: ====[= [ BH555 6 6 BH555Y G G G O , ====[= [ HBHYYYg6668K8KL 4 4 4 EEEE [ E [Aru u % FL!Q25)4v{7H I I I RXq!Q&g 6 6 6 > RXq!Q&g 6 6 6 >  U T T  SSSS  S8888[82222[2III" [899:::::9: [XX77555587[55555rreceZdZdS) Test2DCompatN)rrrr-rrrrsDrr)r)'__doc__numpyrrWpandas.core.dtypes.dtypesrpandasr5pandas._testing_testingr&pandas.api.typesrpandas.core.arrays.numpy_rpandas.tests.extensionrr#rrrfixturer r*r<rArHrNrPrSr\rcr` usefixturesrExtensionTestsreNDArrayBacked2DTestsrr-rrr s|" 222222,,,,,,999999''''''- 3 3S 3C 3 3 3 3 *+++11,+1,FFF 888  4 4 4999    @@@ " " "k%%&9:: ~5~5~5~5~5d1~5~5~5B     4,     r