gL i tddlZddlZddlmZddlmZddlZddl m Z ddl m Z GddejZGdde ZGd d Zd Zd Zej(j+d ej,ej.gdZGddej2j4ZdZy)N)dtypesis_extension_array_dtype)ExtensionArrayc eZdZy) DummyDtypeN)__name__ __module__ __qualname__h/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/pandas/tests/extension/test_common.pyrr sr rc4eZdZddZddZedZddZy) DummyArrayNc||_yNdata)selfrs r__init__zDummyArray.__init__s  r c|jSrrrdtypecopys r __array__zDummyArray.__array__s yyr ctSr)r)rs rrzDummyArray.dtypes |r ct|tr|rt||jS|S|st j ||St j |||S)Nr)rr) isinstancertypernpasarrayarrayrs rastypezDummyArray.astypesP eZ (!tDz$)),,K::d%0 088DD9 9r )returnN)NN)T)r r r rrpropertyrr$r r rrrs% :r rc eZdZejj dejgejgjejejge e e jddggdZejj de jgeje jggdZy)TestExtensionArrayDtypevaluesct|sJyrrrr)s rtest_is_extension_array_dtypez5TestExtensionArrayDtype.test_is_extension_array_dtype(s(///r ct|rJyrrr-s r!test_is_not_extension_array_dtypez9TestExtensionArrayDtype.test_is_not_extension_array_dtype5s+F3333r N)r r r pytestmark parametrizepd CategoricalrSeriesrrr!r#r.r0r r rr(r('s [[ BNN2  BNN2  $ $ BIInbnnR( ) L xrxxA' (   0 0 [[X ibii 6M'NO4P4r r(c"ttjgd}tjgdt}|j t}t j |||j d}t j ||y)Nr*r+robject)rr!r#r:r$tmassert_numpy_array_equal)arrexpectedresults r test_astyper@:sa RXXi( )Cxx 0H ZZ F1 ZZ !F1r cttjgdtj}|j |j d}||usJ|j |j }||usJy)Nr8rF)r)rr!r#int64r$r)r=r?s rtest_astype_no_copyrCEs] RXXirxx8 9C ZZ Z .F &== ZZ "F f  r rcVt|tjsJt|sJyr)rrExtensionDtyperrs rr.r.Os& eV22 33 3 #E ** *r c"eZdZdZfdZxZS)CapturingStringArrayz6Extend StringArray to capture arguments to __getitem__c0||_t| |Sr) last_item_argsuper __getitem__)ritem __class__s rrKz CapturingStringArray.__getitem__Xs!w"4((r )r r r __doc__rK __classcell__)rMs@rrGrGUs@))r rGc tjdttjddgt i}|j dd}|djj}t|dk(sJy)Ncol1helloworldrr*zslice(None, 1, None)) r4 DataFramerGr!r#r:ilocrIstr)df_outs rtest_ellipsis_indexrZ]sk  %bhh/A&PQR B  A V*   ( (C s8- -- -r )numpyr!r1pandas.core.dtypesrpandas.core.dtypes.commonrpandasr4pandas._testing_testingr;pandas.core.arraysrrErrr(r@rCr2r3CategoricalDtype IntervalDtyper.arrays StringArrayrGrZr r rrfs %>- && ::.44&2#:6#:#:#<>Rf>R>R>T"UV+W+ )29900) .r